Spaces:

shanetx
/

shanetx-stable-diffusion-wtx-app

Sleeping

shanetx commited on Dec 31, 2022

Commit

385145e

1 Parent(s): 270bf94

fix data type for cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from diffusers import StableDiffusionImg2ImgPipeline
 device = "cpu"
 model_id_or_path = "runwayml/stable-diffusion-v1-5"
 # model_id_or_path = "shanetx/stable-diffusion-wtx-app"
-img2img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
 img2img_pipe = img2img_pipe.to(device)

 device = "cpu"
 model_id_or_path = "runwayml/stable-diffusion-v1-5"
 # model_id_or_path = "shanetx/stable-diffusion-wtx-app"
+img2img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path)
 img2img_pipe = img2img_pipe.to(device)