Zhibinhong
/

Stable-diffusion-new

Model card Files Files and versions

Zhibinhong commited on Apr 27, 2023

Commit

afa593f

·

1 Parent(s): 97e26aa

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -8,7 +8,7 @@ import json
 class EndpointHandler():
     def __init__(self, path=""):
         model_id = "stabilityai/stable-diffusion-2-1"
-        self.pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
         self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
         self.pipe = self.pipe.to("cuda")

 class EndpointHandler():
     def __init__(self, path=""):
         model_id = "stabilityai/stable-diffusion-2-1"
+        self.pipe = StableDiffusionPipeline.from_pretrained(model_id)
         self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
         self.pipe = self.pipe.to("cuda")