Merge pull request #80 from fenixbao92/main

fix: pytorch => torch in setup.py
2 years ago · ebeb850f22
parent 20bdb5af14 d1e1074995
commit ebeb850f22
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -17,7 +17,7 @@ setup(
        "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={}