Spaces:

smartfeed
/

image2image

Running on Zero

zhiweili commited on Oct 5

Commit

cd21710

•

1 Parent(s): 80b4f1b

change to sd_15

Files changed (1) hide show

app_haircolor_inpaint_15.py CHANGED Viewed

@@ -15,7 +15,7 @@ from diffusers import (
     DDIMScheduler,
 )
-BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-inpainting"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

     DDIMScheduler,
 )
+BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"