Spaces:
Running
on
A10G
Running
on
A10G
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 |
|