Spaces:
Sleeping
Sleeping
Update torch and torchvision for H200 compatible
Browse files- pyproject.toml +1 -1
pyproject.toml
CHANGED
|
@@ -13,7 +13,7 @@ classifiers = [
|
|
| 13 |
"License :: OSI Approved :: Apache Software License",
|
| 14 |
]
|
| 15 |
dependencies = [
|
| 16 |
-
"torch==2.1.
|
| 17 |
"transformers==4.37.2", "tokenizers==0.15.1", "sentencepiece==0.1.99", "shortuuid",
|
| 18 |
"accelerate==0.21.0", "peft==0.4.0", "bitsandbytes==0.41.0",
|
| 19 |
"pydantic<2,>=1", "markdown2[all]", "numpy", "scikit-learn==1.2.2",
|
|
|
|
| 13 |
"License :: OSI Approved :: Apache Software License",
|
| 14 |
]
|
| 15 |
dependencies = [
|
| 16 |
+
"torch==2.1.2", "torchvision==0.16.2", "tiktoken",
|
| 17 |
"transformers==4.37.2", "tokenizers==0.15.1", "sentencepiece==0.1.99", "shortuuid",
|
| 18 |
"accelerate==0.21.0", "peft==0.4.0", "bitsandbytes==0.41.0",
|
| 19 |
"pydantic<2,>=1", "markdown2[all]", "numpy", "scikit-learn==1.2.2",
|