From e281747675412ca79d2f4401bdc31fb88d603c12 Mon Sep 17 00:00:00 2001 From: Robin Huang Date: Mon, 27 Jan 2025 12:53:00 -0800 Subject: [PATCH] Use CUDA 12.4 --- comfy_cli/uv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy_cli/uv.py b/comfy_cli/uv.py index 2a89b3e..3f4449e 100644 --- a/comfy_cli/uv.py +++ b/comfy_cli/uv.py @@ -65,7 +65,7 @@ def parse_req_file(rf: PathLike, skips: Optional[list[str]] = None): class DependencyCompiler: rocmPytorchUrl = "https://download.pytorch.org/whl/rocm6.1" - nvidiaPytorchUrl = "https://download.pytorch.org/whl/cu121" + nvidiaPytorchUrl = "https://download.pytorch.org/whl/cu124" overrideGpu = dedent( """