Update requirements

This commit is contained in:
LIU, Zichen
2024-12-08 17:21:16 +08:00
committed by GitHub
parent 4ba3c01fae
commit 403abf6036
+1 -1
View File
@@ -15,7 +15,7 @@ classifiers = [
dependencies = [
"torch==2.1.2", "torchvision==0.16.2",
"transformers==4.37.2", "tokenizers==0.15.1", "sentencepiece==0.2.0", "shortuuid",
"accelerate==0.33.0", "peft", "bitsandbytes",
"accelerate==0.33.0", "peft==0.13.2", "bitsandbytes",
"pydantic", "markdown2[all]", "numpy", "scikit-learn==1.2.2",
"gradio==5.4.0", "gradio_client==1.4.2", "requests", "httpx==0.24.1", "uvicorn", "fastapi",
"einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13",