Spaces:

Ashoka74
/

Demo_Refurnish

Running on Zero

Ashoka74 commited on 18 days ago

Commit

85d7e1e

•

1 Parent(s): cd949a9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -71,8 +71,7 @@ vae = AutoencoderKL.from_pretrained(sd15_name, subfolder="vae")
 unet = UNet2DConditionModel.from_pretrained(sd15_name, subfolder="unet")
 # Load model directly
 from transformers import AutoModelForImageSegmentation
-rmbg = AutoModelForImageSegmentation.from_pretrained("briaai/RMBG-2.0", trust_remote_code=True)
 model = DepthAnythingV2(encoder='vits', features=64, out_channels=[48, 96, 192, 384])
 model.load_state_dict(torch.load('checkpoints/depth_anything_v2_vits.pth', map_location=device))

 unet = UNet2DConditionModel.from_pretrained(sd15_name, subfolder="unet")
 # Load model directly
 from transformers import AutoModelForImageSegmentation
+rmbg = AutoModelForImageSegmentation.from_pretrained("briaai/RMBG-1.4", trust_remote_code=True)
 model = DepthAnythingV2(encoder='vits', features=64, out_channels=[48, 96, 192, 384])
 model.load_state_dict(torch.load('checkpoints/depth_anything_v2_vits.pth', map_location=device))