Spaces:

yigitekin
/

CLIPAway

Running on Zero

yigitekin commited on Sep 27

Commit

02ab530

•

1 Parent(s): 9033476

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ login()
 # Load configuration and models
 config = OmegaConf.load("config/inference_config.yaml")
 sd_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
-    "benjamin-paine/stable-diffusion-v1-5-inpainting", torch_dtype=torch.float32
 )
 clipaway = CLIPAway(
     sd_pipe=sd_pipeline,

 # Load configuration and models
 config = OmegaConf.load("config/inference_config.yaml")
 sd_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
+    "botp/stable-diffusion-v1-5-inpainting", torch_dtype=torch.float32
 )
 clipaway = CLIPAway(
     sd_pipe=sd_pipeline,