viba98 commited on
Commit
4a8b43e
·
1 Parent(s): 10b71ad

Update handler.py

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