diff --git a/services/comfy_ui.py b/services/comfy_ui.py index a8ef5db..b9b36d3 100644 --- a/services/comfy_ui.py +++ b/services/comfy_ui.py @@ -19,6 +19,7 @@ class ComfyUi(Stack): self.git_clone(url="https://github.com/ltdrdata/ComfyUI-Manager.git", dest="webui/custom_nodes/manager") # Add GGUF support + self.git_clone(url="https://github.com/city96/ComfyUI-GGUF.git", dest="webui/custom_nodes/ComfyUI-GGUF") self.pip_install(["gguf", "numpy==1.26.4"]) # Add NF4 support for Flux @@ -27,6 +28,6 @@ class ComfyUi(Stack): dest="webui/custom_nodes/ComfyUI_bitsandbytes_NF4") def _start(self): - args = ["--port", str(self.port)] + args = ["--port", str(self.port), "--force-fp32"] self.python(f"main.py", args=args, current_dir="webui", env=["TORCH_BLAS_PREFER_HIPBLASLT=0"], daemon=True)