diff --git a/pyproject.toml b/pyproject.toml index ef3833a..578ebd4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,6 +14,7 @@ dependencies = [ "tiktoken>=0.11.0", "tokenizers>=0.22.0", "torch>=2.8.0", + "transformers>=4.0.0", "uvicorn>=0.36.0", "wandb>=0.21.3", ]