Skip to content

Commit

Permalink
Update torchvision version to 0.16.0
Browse files Browse the repository at this point in the history
  • Loading branch information
ZhangYuanhan committed Mar 8, 2024
1 parent 58cf7ec commit dc6c50d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ classifiers = [
]

dependencies = [
"torch==2.1.0", "torchvision==0.15.2",
"torch==2.1.0", "torchvision==0.16.0",
"transformers==4.38.2", "tokenizers==0.15.2", "sentencepiece==0.1.99", "shortuuid",
"accelerate==0.27.2", "peft==0.4.0", "bitsandbytes==0.41.0",
"pydantic<2,>=1", "markdown2[all]", "numpy", "scikit-learn==1.2.2",
Expand Down

0 comments on commit dc6c50d

Please sign in to comment.