KawaiiApp commited on
Commit
6ad32f2
1 Parent(s): 966bbc8

path removed

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -17,7 +17,7 @@ if device.type != 'cuda':
17
  raise ValueError("need to run on GPU")
18
 
19
  class EndpointHandler():
20
- def __init__(self, path="tomriddle/anythinv3-vae"):
21
  # load the optimized model
22
  self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16,low_cpu_mem_usage=False)
23
  self.pipe = self.pipe.to(device)
 
17
  raise ValueError("need to run on GPU")
18
 
19
  class EndpointHandler():
20
+ def __init__(self, path=""):
21
  # load the optimized model
22
  self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16,low_cpu_mem_usage=False)
23
  self.pipe = self.pipe.to(device)