KawaiiApp
/

anythinv3-vae-handler

StableDiffusionPipeline

stable-diffusion

stable-diffusion-diffusers

Model card Files Files and versions

KawaiiApp commited on Jun 28, 2023

Commit

35c6947

·

1 Parent(s): 0180dcb

added to pipe

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -20,7 +20,7 @@ class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
         self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16,low_cpu_mem_usage=False)
-        self.pipeline.scheduler = EulerDiscreteScheduler.from_config(self.pipeline.scheduler.config)
         self.pipe = self.pipe.to(device)

     def __init__(self, path=""):
         # load the optimized model
         self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16,low_cpu_mem_usage=False)
+        self.pipe.scheduler = EulerDiscreteScheduler.from_config(self.pipe.scheduler.config)
         self.pipe = self.pipe.to(device)