diff --git a/pyproject.toml b/pyproject.toml index 6dd885b..d92a72b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,7 +13,7 @@ classifiers = [ ] dependencies = [ "torch==2.2.0", "torchvision==0.17.0", - "transformers==4.45.0", "deepspeed==0.15.1", "tokenizers==0.20.3", "sentencepiece==0.1.99", "shortuuid", + "transformers==4.45.0", "deepspeed==0.15.1", "tokenizers==0.20.3", "sentencepiece==0.2.1", "shortuuid", "accelerate==0.34.2", "decord==0.6.0", "peft", "bitsandbytes", "pydantic", "markdown2[all]", "numpy", "scikit-learn==1.2.2", "gradio==4.19.2", "gradio_client==0.10.1",