fix: pytorch => torch in setup.py

2 years ago · d1e1074995
parent 394df4187d
commit d1e1074995
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={}