kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

print model dir

Browse Source
This commit is contained in:
Qing 2024-02-05 21:12:49 +08:00
parent 75b233023c
commit 93f965da40
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
iopaint/runtime.py
View File

@ -79,6 +79,7 @@ def check_device(device: Device) -> Device:
def setup_model_dir(model_dir: Path):
model_dir = model_dir.expanduser().absolute()
logger.info(f"Model directory: {model_dir}")
os.environ["U2NET_HOME"] = str(model_dir)
os.environ["XDG_CACHE_HOME"] = str(model_dir)
if not model_dir.exists():