Spaces:
Runtime error
Runtime error
Linoy Tsaban
commited on
Commit
·
5e2c5e4
1
Parent(s):
2a0f200
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ from share_btn import community_icon_html, loading_icon_html, share_js
|
|
| 18 |
sd_model_id = "runwayml/stable-diffusion-v1-5"
|
| 19 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 20 |
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
|
| 21 |
-
pipe = SemanticStableDiffusionImg2ImgPipeline_DPMSolver.from_pretrained(sd_model_id,vae=vae,torch_dtype=torch.float16, safety_checker=None,
|
| 22 |
pipe.scheduler = DPMSolverMultistepSchedulerInject.from_pretrained(sd_model_id, subfolder="scheduler"
|
| 23 |
, algorithm_type="sde-dpmsolver++", solver_order=2)
|
| 24 |
|
|
|
|
| 18 |
sd_model_id = "runwayml/stable-diffusion-v1-5"
|
| 19 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 20 |
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
|
| 21 |
+
pipe = SemanticStableDiffusionImg2ImgPipeline_DPMSolver.from_pretrained(sd_model_id,vae=vae,torch_dtype=torch.float16, safety_checker=None, requires_safety_checker=False).to(device)
|
| 22 |
pipe.scheduler = DPMSolverMultistepSchedulerInject.from_pretrained(sd_model_id, subfolder="scheduler"
|
| 23 |
, algorithm_type="sde-dpmsolver++", solver_order=2)
|
| 24 |
|