Spaces:

mPLUG
/

TinyChart-3B

Running on Zero

xzl12306 commited on about 11 hours ago

Commit

6affa5e

verified ·

1 Parent(s): 15f9a43

Update torch and torchvision for H200 compatible

Files changed (1) hide show

pyproject.toml CHANGED Viewed

@@ -13,7 +13,7 @@ classifiers = [
     "License :: OSI Approved :: Apache Software License",
 ]
 dependencies = [
-    "torch==2.1.0", "torchvision==0.16.0", "tiktoken",
     "transformers==4.37.2", "tokenizers==0.15.1", "sentencepiece==0.1.99", "shortuuid",
     "accelerate==0.21.0", "peft==0.4.0", "bitsandbytes==0.41.0",
     "pydantic<2,>=1", "markdown2[all]", "numpy", "scikit-learn==1.2.2",

     "License :: OSI Approved :: Apache Software License",
 ]
 dependencies = [
+    "torch==2.1.2", "torchvision==0.16.2", "tiktoken",
     "transformers==4.37.2", "tokenizers==0.15.1", "sentencepiece==0.1.99", "shortuuid",
     "accelerate==0.21.0", "peft==0.4.0", "bitsandbytes==0.41.0",
     "pydantic<2,>=1", "markdown2[all]", "numpy", "scikit-learn==1.2.2",