Omnibus commited on
Commit
ff53b96
1 Parent(s): 03af647

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