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):