viba98 commited on
Commit
341c7be
·
1 Parent(s): dc16fab

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -13,7 +13,7 @@ if device.type != 'cuda':
13
  raise ValueError("need to run on GPU")
14
 
15
  class EndpointHandler():
16
- def __init__(self, path=""):
17
  # load the optimized model
18
  self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
19
  self.pipe = self.pipe.to(device)
 
13
  raise ValueError("need to run on GPU")
14
 
15
  class EndpointHandler():
16
+ def __init__(self, path="vmanot/valiant-effort-one"):
17
  # load the optimized model
18
  self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
19
  self.pipe = self.pipe.to(device)