KawaiiApp commited on
Commit
35c6947
1 Parent(s): 0180dcb

added to pipe

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -20,7 +20,7 @@ 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.pipeline.scheduler = EulerDiscreteScheduler.from_config(self.pipeline.scheduler.config)
24
  self.pipe = self.pipe.to(device)
25
 
26
 
 
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.scheduler = EulerDiscreteScheduler.from_config(self.pipe.scheduler.config)
24
  self.pipe = self.pipe.to(device)
25
 
26