Spaces:
Runtime error
Runtime error
Update video_diffusion/inpaint_zoom/zoom_in_app.py
Browse files
video_diffusion/inpaint_zoom/zoom_in_app.py
CHANGED
@@ -35,7 +35,9 @@ class StableDiffusionZoomIn:
|
|
35 |
|
36 |
def load_model(self, model_id):
|
37 |
if self.pipe is None:
|
38 |
-
self.pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=dtype, revision="fp16")
|
|
|
|
|
39 |
self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
|
40 |
self.pipe = self.pipe.to(device)
|
41 |
self.pipe.safety_checker = dummy
|
|
|
35 |
|
36 |
def load_model(self, model_id):
|
37 |
if self.pipe is None:
|
38 |
+
#self.pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=dtype, revision="fp16")
|
39 |
+
self.pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=dtype)
|
40 |
+
|
41 |
self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
|
42 |
self.pipe = self.pipe.to(device)
|
43 |
self.pipe.safety_checker = dummy
|