From 74c63f6123654b1bdb754fd143c25cbb13c2bd1a Mon Sep 17 00:00:00 2001
From: AI Lab <129358391+1038lab@users.noreply.github.com>
Date: Mon, 7 Jul 2025 20:52:39 -0700
Subject: [PATCH] Update AILab_BiRefNet.py

---
 AILab_BiRefNet.py | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/AILab_BiRefNet.py b/AILab_BiRefNet.py
index f1320f0..f1ba50c 100644
--- a/AILab_BiRefNet.py
+++ b/AILab_BiRefNet.py
@@ -141,6 +141,34 @@ MODEL_CONFIG = {
         "default_res": 2048,
         "max_res": 2560,
         "min_res": 1024
+    },
+    "BiRefNet_dynamic": {
+        "repo_id": "1038lab/BiRefNet_dynamic",
+        "files": {
+            "birefnet.py": "birefnet.py",
+            "BiRefNet_config.py": "BiRefNet_config.py",
+            "model.safetensors": "model.safetensors",
+            "config.json": "config.json"
+        },
+        "cache_dir": "BiRefNet_dynamic",
+        "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",
+        "files": {
+            "birefnet_lite.py": "birefnet_lite.py",
+            "BiRefNet_config.py": "BiRefNet_config.py",
+            "model.safetensors": "model.safetensors",
+            "config.json": "config.json"
+        },
+        "cache_dir": "BiRefNet_lite-matting",
+        "description": "Lightweight matting model for general purpose",
+        "default_res": 1024,
+        "max_res": 2048,
+        "min_res": 512
     }
 }
 
@@ -438,4 +466,4 @@ NODE_CLASS_MAPPINGS = {
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "BiRefNetRMBG": "BiRefNet (RMBG)"
-}
\ No newline at end of file
+}