Remove mandatory pytorch requirement

paul-grundmann · Aug 14, 2024 · 422b889 · 422b889
1 parent d8a4488
commit 422b889
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements-cuda.txt b/requirements-cuda.txt
@@ -4,7 +4,7 @@
 # Dependencies for NVIDIA GPUs
 ray >= 2.9
 nvidia-ml-py # for pynvml package
-torch == 2.4.0
+#torch == 2.4.0
 # These must be updated alongside torch
 torchvision == 0.19   # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
 xformers == 0.0.27.post2; platform_system == 'Linux' and platform_machine == 'x86_64'  # Requires PyTorch 2.4.0