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
|