patrickvonplaten commited on
Commit
4031105
1 Parent(s): fc76896
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -27,9 +27,7 @@ class Model:
27
  self.pipe_t2i.scheduler = DPMSolverMultistepScheduler.from_config(
28
  self.pipe_t2i.scheduler.config
29
  )
30
- self.pipe_i2i = StableDiffusionImg2ImgPipeline(
31
- **self.pipe_t2i.components, torch_dtype=torch.float16
32
- )
33
  else:
34
  self.pipe_t2i = None
35
  self.pipe_i2i = None
 
27
  self.pipe_t2i.scheduler = DPMSolverMultistepScheduler.from_config(
28
  self.pipe_t2i.scheduler.config
29
  )
30
+ self.pipe_i2i = StableDiffusionImg2ImgPipeline(**self.pipe_t2i.components)
 
 
31
  else:
32
  self.pipe_t2i = None
33
  self.pipe_i2i = None