diff --git a/pyproject.toml b/pyproject.toml index 13b9751..6e84365 100755 --- a/pyproject.toml +++ b/pyproject.toml @@ -16,10 +16,10 @@ dependencies = [ "torch==2.1.2", "torchvision==0.16.2", "transformers==4.43.1", "tokenizers==0.19.0", "sentencepiece==0.1.99", "shortuuid", "accelerate==0.29.0", "peft", "bitsandbytes", - "pydantic", "markdown2[all]", "numpy", "scikit-learn==1.2.2", + "pydantic", "markdown2[all]", "numpy==1.26.4", "scikit-learn==1.2.2", "gradio==5.9.1", "gradio_client==1.5.2", "requests", "httpx==0.28.1", "uvicorn", "fastapi", - "einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13", + "einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13", "decord==0.6.0", ] [project.optional-dependencies]