Update tasks/ref_in.py

tasks/ref_in.py

@@ -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",
+    "stabilityai/stable-diffusion-2-inpainting",
+    # "runwayml/stable-diffusion-inpainting",
     revision="fp16", 
     torch_dtype=torch.float16,
 ).to("cuda")