patrickvonplaten commited on
Commit
b242a38
1 Parent(s): cbd23ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ class Model:
30
  self.pipe_t2i.scheduler = DPMSolverMultistepScheduler.from_config(
31
  self.pipe_t2i.scheduler.config
32
  )
33
- self.pipe_i2i = StableDiffusionImg2ImgPipeline(**self.pipe_t2i.components, safety_checker=SAFETY_CHECKER)
34
  else:
35
  self.pipe_t2i = None
36
  self.pipe_i2i = None
 
30
  self.pipe_t2i.scheduler = DPMSolverMultistepScheduler.from_config(
31
  self.pipe_t2i.scheduler.config
32
  )
33
+ self.pipe_i2i = StableDiffusionImg2ImgPipeline(**self.pipe_t2i.components)
34
  else:
35
  self.pipe_t2i = None
36
  self.pipe_i2i = None