From 40595d1a9a407dafd393ab20bf83612d7b0b7474 Mon Sep 17 00:00:00 2001
From: AI Lab <129358391+1038lab@users.noreply.github.com>
Date: Fri, 11 Jul 2025 09:11:13 -0700
Subject: [PATCH] Add files via upload

---
 AILab_BiRefNet.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/AILab_BiRefNet.py b/AILab_BiRefNet.py
index f1ba50c..29800f9 100644
--- a/AILab_BiRefNet.py
+++ b/AILab_BiRefNet.py
@@ -143,28 +143,28 @@ MODEL_CONFIG = {
         "min_res": 1024
     },
     "BiRefNet_dynamic": {
-        "repo_id": "1038lab/BiRefNet_dynamic",
+        "repo_id": "1038lab/BiRefNet",
         "files": {
             "birefnet.py": "birefnet.py",
             "BiRefNet_config.py": "BiRefNet_config.py",
-            "model.safetensors": "model.safetensors",
+            "BiRefNet_dynamic.safetensors": "BiRefNet_dynamic.safetensors",
             "config.json": "config.json"
         },
-        "cache_dir": "BiRefNet_dynamic",
+        "cache_dir": "BiRefNet",
         "description": "Dynamic model for high-resolution dichotomous image segmentation",
         "default_res": 1024,
         "max_res": 2048,
         "min_res": 512
     },
     "BiRefNet_lite-matting": {
-        "repo_id": "1038lab/BiRefNet_lite-matting",
+        "repo_id": "1038lab/BiRefNet",
         "files": {
             "birefnet_lite.py": "birefnet_lite.py",
             "BiRefNet_config.py": "BiRefNet_config.py",
-            "model.safetensors": "model.safetensors",
+            "BiRefNet_lite-matting.safetensors": "BiRefNet_lite-matting.safetensors",
             "config.json": "config.json"
         },
-        "cache_dir": "BiRefNet_lite-matting",
+        "cache_dir": "BiRefNet",
         "description": "Lightweight matting model for general purpose",
         "default_res": 1024,
         "max_res": 2048,
@@ -466,4 +466,4 @@ NODE_CLASS_MAPPINGS = {
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "BiRefNetRMBG": "BiRefNet (RMBG)"
-}
+}
\ No newline at end of file