Spaces:
Running
Running
Update app.py
Browse files
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:
|