fix: serve from UI has no workspace (#45)

Signed-off-by: Frost Ming <[email protected]>
bentoml · Feb 12, 2025 · d5dbda8 · d5dbda8
1 parent cbf66db
commit d5dbda8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/comfy_pack/service.py b/src/comfy_pack/service.py
@@ -122,7 +122,7 @@ def generate(
             output_dir=ctx.temp_dir,
             timeout=REQUEST_TIMEOUT,
             verbose=verbose,
-            workspace=self.server.workspace,
+            workspace=self.server.workspace if hasattr(self, "server") else ".",
             **kwargs,
         )
         if isinstance(ret, list):