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