print model dir

This commit is contained in:
Qing 2024-02-05 21:12:49 +08:00
parent 75b233023c
commit 93f965da40

View File

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