From dd2fd04fd7f5dd0fee2e683bff5ac2aab6c4c32c Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Fri, 26 Jan 2024 11:17:58 +0100 Subject: [PATCH] feat: set U2NET_HOME env var to path_inpaint previously was using the user dir, see https://github.com/danielgatis/rembg/blob/49d1686f65b71b72df1c3b420551406d13074426/rembg/sessions/base.py#L78 --- launch.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/launch.py b/launch.py index e98045f6..63a38767 100644 --- a/launch.py +++ b/launch.py @@ -22,8 +22,9 @@ from build_launcher import build_launcher from modules.launch_util import is_installed, run, python, run_pip, requirements_met from modules.model_loader import load_file_from_url from modules.config import path_checkpoints, path_loras, path_vae_approx, path_fooocus_expansion, \ - checkpoint_downloads, path_embeddings, embeddings_downloads, lora_downloads + path_inpaint, checkpoint_downloads, path_embeddings, embeddings_downloads, lora_downloads +os.environ["U2NET_HOME"] = path_inpaint REINSTALL_ALL = False TRY_INSTALL_XFORMERS = False