Zyrenth commited on
Commit
2e82b54
·
verified ·
1 Parent(s): bf24c2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -114,7 +114,7 @@ else: # download all models
114
  vae=vae,
115
  torch_dtype=dtype,
116
  scheduler=scheduler,
117
- safety_checker=None,
118
  )
119
  model.pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
120
  model.path,
@@ -122,7 +122,7 @@ else: # download all models
122
  vae=vae,
123
  torch_dtype=dtype,
124
  scheduler=scheduler,
125
- safety_checker=None,
126
  )
127
  except Exception as e:
128
  print(
 
114
  vae=vae,
115
  torch_dtype=dtype,
116
  scheduler=scheduler,
117
+ # safety_checker=None,
118
  )
119
  model.pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
120
  model.path,
 
122
  vae=vae,
123
  torch_dtype=dtype,
124
  scheduler=scheduler,
125
+ # safety_checker=None,
126
  )
127
  except Exception as e:
128
  print(