From c9c30cf3bbcfea3c2a800c9effcd2eb0549bd3fb Mon Sep 17 00:00:00 2001
From: catboxanon <122327233+catboxanon@users.noreply.github.com>
Date: Tue, 30 May 2023 00:26:12 +0000
Subject: [PATCH] Bump to 2.0.38, support DIS

---
 install.py                      | 2 +-
 scripts/postprocessing_rembg.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/install.py b/install.py
index 5c13936..1a73e3c 100644
--- a/install.py
+++ b/install.py
@@ -1,7 +1,7 @@
 import launch
 
 if not launch.is_installed("rembg"):
-    launch.run_pip("install rembg==2.0.30 --no-deps", "rembg")
+    launch.run_pip("install rembg==2.0.38 --no-deps", "rembg")
 
 for dep in ['onnxruntime', 'pymatting', 'pooch']:
     if not launch.is_installed(dep):
diff --git a/scripts/postprocessing_rembg.py b/scripts/postprocessing_rembg.py
index 50b3fec..8d64332 100644
--- a/scripts/postprocessing_rembg.py
+++ b/scripts/postprocessing_rembg.py
@@ -11,6 +11,8 @@ models = [
     "u2net_human_seg",
     "u2net_cloth_seg",
     "silueta",
+    "isnet-general-use",
+    "isnet-anime",
 ]
 
 class ScriptPostprocessingUpscale(scripts_postprocessing.ScriptPostprocessing):