Hugging Face's logo

Spaces:
xdecoder
/
Demo
Runtime error

App Files Community
5

Update tasks/ref_in.py

#5
by mohan007 - opened
base: refs/heads/main
←
from: refs/pr/5
Discussion Files changed
+2
-2
Files changed (1) hide show
  1. tasks/ref_in.py +2 -2
tasks/ref_in.py CHANGED
@@ -23,8 +23,8 @@ transform = transforms.Compose(t)
23
  metadata = MetadataCatalog.get('ade20k_panoptic_train')
24
 
25
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
26
- # "stabilityai/stable-diffusion-2-inpainting",
27
- "runwayml/stable-diffusion-inpainting",
28
  revision="fp16",
29
  torch_dtype=torch.float16,
30
  ).to("cuda")
 
23
  metadata = MetadataCatalog.get('ade20k_panoptic_train')
24
 
25
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
26
+ "stabilityai/stable-diffusion-2-inpainting",
27
+ # "runwayml/stable-diffusion-inpainting",
28
  revision="fp16",
29
  torch_dtype=torch.float16,
30
  ).to("cuda")