Pin the pytorch version for managed server install

- at least 2.4.0 seems to cause problems #966
Acly · Jul 24, 2024 · c7ac5e1 · c7ac5e1
1 parent 1d02919
commit c7ac5e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ai_diffusion/server.py b/ai_diffusion/server.py
@@ -186,7 +186,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", "torchvision", "torchaudio"]
+        torch_args = ["torch==2.3.1", "torchvision==0.18.1", "torchaudio==2.3.1"]
         if self.backend is ServerBackend.cpu or self.backend is ServerBackend.mps:
             torch_args += ["--index-url", "https://download.pytorch.org/whl/cpu"]
         elif self.backend is ServerBackend.cuda: