Merge pull request #40 from replicate/yorickvp/fix-cog-build

Remove superfluous torch==2.4.1 dependency
replicate · Nov 1, 2024 · 83c440c · 83c440c
2 parents 1d75bdd + 6f4fe19
commit 83c440c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/cog.yaml.template b/cog.yaml.template
@@ -10,7 +10,6 @@ build:
 
   python_packages:
     - "numpy<2"
-    - "torch==2.4.1"
     - "einops==0.8.0"
     - "fire==0.6.0"
     - "huggingface-hub==0.24.3"