Spaces:

xdecoder
/

Demo

Runtime error

mohan007 commited on Sep 4, 2023

Commit

af74e16

•

1 Parent(s): 5e669f7

Update tasks/ref_in.py

Files changed (1) hide show

tasks/ref_in.py CHANGED Viewed

@@ -23,8 +23,8 @@ transform = transforms.Compose(t)
 metadata = MetadataCatalog.get('ade20k_panoptic_train')
 pipe = StableDiffusionInpaintPipeline.from_pretrained(
-    # "stabilityai/stable-diffusion-2-inpainting",
-    "runwayml/stable-diffusion-inpainting",
     revision="fp16",
     torch_dtype=torch.float16,
 ).to("cuda")

 metadata = MetadataCatalog.get('ade20k_panoptic_train')
 pipe = StableDiffusionInpaintPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-2-inpainting",
+    # "runwayml/stable-diffusion-inpainting",
     revision="fp16",
     torch_dtype=torch.float16,
 ).to("cuda")