fix: specify the workspace for run (#38)

Signed-off-by: Frost Ming <[email protected]>
bentoml · Jan 24, 2025 · 3f07d59 · 3f07d59
1 parent e181b07
commit 3f07d59
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/src/comfy_pack/cli.py b/src/comfy_pack/cli.py
@@ -404,6 +404,7 @@ def run(ctx, cpack: str, output_dir: str, help: bool, verbose: int):
             workflow,
             Path(output_dir).absolute(),
             verbose=verbose,
+            workspace=server.workspace,
             **validated_data.model_dump(),
         )
         console.print("\n[green]✓ Workflow is executed successfully![/green]")

diff --git a/src/comfy_pack/run.py b/src/comfy_pack/run.py
@@ -206,6 +206,7 @@ def run_workflow(
     output_dir: Union[str, Path, None] = None,
     timeout: int = 300,
     verbose: int = 0,
+    workspace: str = ".",
     **kwargs: Any,
 ) -> Any:
     """
@@ -258,6 +259,8 @@ def run_workflow(
     command = [
         "comfy",
         "--skip-prompt",
+        "--workspace",
+        workspace,
         "run",
         "--workflow",
         workflow_file_path.as_posix(),

diff --git a/src/comfy_pack/service.py b/src/comfy_pack/service.py
@@ -73,7 +73,7 @@ def _watch_server(server: comfy_pack.run.ComfyUIServer):
     snapshot = {}
 
 
-@bentoml.mount_asgi_app(app, path="/comfy")
+@bentoml.asgi_app(app, path="/comfy")
 @bentoml.service(traffic={"timeout": REQUEST_TIMEOUT * 2}, resources={"gpu": 1})
 class ComfyService:
     def __init__(self):
@@ -122,6 +122,7 @@ def generate(
             output_dir=ctx.temp_dir,
             timeout=REQUEST_TIMEOUT,
             verbose=verbose,
+            workspace=self.server.workspace,
             **kwargs,
         )
         if isinstance(ret, list):