Merge pull request THUDM#80 from fenixbao92/main

fix: pytorch => torch in setup.py
kester2015 · Mar 6, 2023 · ebeb850 · ebeb850
2 parents 20bdb5a + d1e1074
commit ebeb850
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -17,7 +17,7 @@
         "setuptools>=58.0.4",
         "transformers>=4.22.0,<=4.24.0",
         "tokenizers>=0.11.0,<=0.11.4",
-        "pytorch>=1.10.0",
+        "torch>=1.10.0",
         "tqdm>=4.63.0",
     ],
     entry_points={}