Don't pin pytorch version (#6)

2.0.1 is no longer available
appeler · Nov 29, 2024 · 8f38a36 · 8f38a36
1 parent d929e8a
commit 8f38a36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 pandas>=1.3.0
 numpy
 joblib==1.3.1
-torch==2.0.1
+torch>=2.0.1
 tqdm==4.66.3
 scikit-learn==1.3.0