Linoy Tsaban commited on
Commit
2a0f200
1 Parent(s): 7c80d27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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).to(device)
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, require_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