FreeSplatter

Runtime error

bluestyle97 commited on Dec 19, 2024

Commit

944d7dc

verified ·

1 Parent(s): 1d262d9

Update freesplatter/webui/runner.py

Files changed (1) hide show

freesplatter/webui/runner.py CHANGED Viewed

@@ -75,7 +75,8 @@ class FreeSplatterRunner:
         # background remover
         self.rembg = AutoModelForImageSegmentation.from_pretrained(
-            "ZhengPeng7/BiRefNet",
             trust_remote_code=True,
         ).to(device)
         self.rembg.eval()

         # background remover
         self.rembg = AutoModelForImageSegmentation.from_pretrained(
+            # "ZhengPeng7/BiRefNet",
+            "briaai/RMBG-2.0",
             trust_remote_code=True,
         ).to(device)
         self.rembg.eval()