Spaces:

yizhangliu
/

Grounded-Segment-Anything

Running on T4

sigyllly commited on Feb 2

Commit

f861ca4

•

1 Parent(s): 35bdba8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -320,7 +320,7 @@ def load_sd_model(device):
     sd_model = None
     if os.environ.get('IS_MY_DEBUG') is None:
         sd_model = StableDiffusionInpaintPipeline.from_pretrained(
-                "runwayml/stable-diffusion-inpainting",
                 revision="fp16",
                 # "stabilityai/stable-diffusion-2-inpainting",
                 torch_dtype=torch.float16,

     sd_model = None
     if os.environ.get('IS_MY_DEBUG') is None:
         sd_model = StableDiffusionInpaintPipeline.from_pretrained(
+                "Lykon/absolute-reality-1.6525-inpainting",
                 revision="fp16",
                 # "stabilityai/stable-diffusion-2-inpainting",
                 torch_dtype=torch.float16,