feat: set U2NET_HOME env var to path_inpaint

previously was using the user dir, see 49d1686f65/rembg/sessions/base.py (L78)
This commit is contained in:
Manuel Schmid 2024-01-26 11:17:58 +01:00
parent 62fe86f1e8
commit dd2fd04fd7
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 2 additions and 1 deletions

View File

@ -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