From d15c4c54add95c07cd2e55101a2dea772e200b73 Mon Sep 17 00:00:00 2001 From: Frost Ming Date: Tue, 4 Feb 2025 19:18:53 +0800 Subject: [PATCH] fix: don't import bentoml at setup (#41) Signed-off-by: Frost Ming --- src/comfy_pack/setup_workspace.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/comfy_pack/setup_workspace.py b/src/comfy_pack/setup_workspace.py index b744e20..33d1be1 100755 --- a/src/comfy_pack/setup_workspace.py +++ b/src/comfy_pack/setup_workspace.py @@ -22,12 +22,12 @@ def _get_workspace() -> tuple[Path, dict]: import hashlib import json - from bentoml._internal.configuration.containers import BentoMLContainer + bentoml_home = os.getenv( + "BENTOML_HOME", os.path.join(os.path.expanduser("~"), "bentoml") + ) checksum = hashlib.md5(SNAPSHOT.strip().encode("utf8")).hexdigest() - wp = ( - Path(BentoMLContainer.bentoml_home.get()) / "run" / "comfy_workspace" / checksum - ) + wp = Path(bentoml_home) / "run" / "comfy_workspace" / checksum wp.parent.mkdir(parents=True, exist_ok=True) return wp, json.loads(SNAPSHOT)