r3gm commited on
Commit
0a0d059
1 Parent(s): 16b8689

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a
37
 
38
  pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
39
  pipe.to(torch_device="cpu", torch_dtype=DTYPE)
40
-
41
 
42
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
43
  if randomize_seed:
 
37
 
38
  pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
39
  pipe.to(torch_device="cpu", torch_dtype=DTYPE)
40
+ pipe.safety_checker = None
41
 
42
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
43
  if randomize_seed: