Update managed install to pytorch 2.5.1, update Comfy/nodes to latest…

… versions
Acly · Jan 15, 2025 · e2b055c · e2b055c
1 parent 15b4f62
commit e2b055c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/ai_diffusion/resources.py b/ai_diffusion/resources.py
@@ -26,21 +26,21 @@ class CustomNode(NamedTuple):
         "ControlNet Preprocessors",
         "comfyui_controlnet_aux",
         "https://github.com/Fannovel16/comfyui_controlnet_aux",
-        "6c563c5032f77dd3336b603bde3d12b415d003ab",
+        "5a049bde9cc117dafc327cded156459289097ea1",
         ["InpaintPreprocessor", "DepthAnythingV2Preprocessor"],
     ),
     CustomNode(
         "IP-Adapter",
         "ComfyUI_IPAdapter_plus",
         "https://github.com/cubiq/ComfyUI_IPAdapter_plus",
-        "88a71407c545e4eb0f223294f5b56302ef8696f3",
+        "b188a6cb39b512a9c6da7235b880af42c78ccd0d",
         ["IPAdapterModelLoader", "IPAdapter"],
     ),
     CustomNode(
         "External Tooling Nodes",
         "comfyui-tooling-nodes",
         "https://github.com/Acly/comfyui-tooling-nodes",
-        "50d3479fba55116334ed9fb1ad15f13a9294badf",
+        "8ed5591574ff6340729970eed780e79401568dad",
         ["ETN_LoadImageBase64", "ETN_LoadMaskBase64", "ETN_SendImageWebSocket", "ETN_Translate"],
     ),
     CustomNode(
@@ -57,7 +57,7 @@ class CustomNode(NamedTuple):
         "GGUF",
         "ComfyUI-GGUF",
         "https://github.com/city96/ComfyUI-GGUF",
-        "4a8432884167f2526d60ef36e985bdabebb9e1e0",
+        "5875c52f59baca3a9372d68c43a3775e21846fe0",
         ["UnetLoaderGGUF", "DualCLIPLoaderGGUF"],
     )
 ]

diff --git a/ai_diffusion/server.py b/ai_diffusion/server.py
@@ -196,7 +196,7 @@ async def _install_comfy(self, comfy_dir: Path, network: QNetworkAccessManager,
         await _extract_archive("ComfyUI", archive_path, comfy_dir.parent, cb)
         temp_comfy_dir = comfy_dir.parent / f"ComfyUI-{resources.comfy_version}"
 
-        torch_args = ["torch~=2.4.1", "torchvision~=0.19.1", "torchaudio~=2.4.1"]
+        torch_args = ["torch~=2.5.1", "torchvision~=0.20.1", "torchaudio~=2.5.1"]
         if self.backend is ServerBackend.cpu:
             torch_args += ["--index-url", "https://download.pytorch.org/whl/cpu"]
         elif self.backend is ServerBackend.cuda: