diff --git a/requirements.txt b/requirements.txt index 4904b422a90..257b3ccbd72 100644 --- a/requirements.txt +++ b/requirements.txt @@ -307,7 +307,7 @@ torch-fidelity==0.3.0 torchmetrics==0.11.4 torchvision~=0.16.2 tqdm==4.66.4 -transformers==4.41.2 +transformers==5.5.0 transformers-stream-generator==0.0.5 trio==0.25.1 trio-websocket==0.11.1 diff --git a/setup.cfg b/setup.cfg index 8416581de21..5ba28ed554d 100644 --- a/setup.cfg +++ b/setup.cfg @@ -53,7 +53,7 @@ install_requires= scikit-learn~=1.1 # Models and Metrics Extras - transformers~=4.40 # For anthropic_client, vision_language.huggingface_vlm_client, huggingface_client, huggingface_tokenizer, test_openai_token_cost_estimator, model_summac (via summarization_metrics) + transformers~=5.5 # For anthropic_client, vision_language.huggingface_vlm_client, huggingface_client, huggingface_tokenizer, test_openai_token_cost_estimator, model_summac (via summarization_metrics) # TODO: Upgrade torch - we need > 2.0.0 for newer versions of transformers torch>=1.13.1,<3.0.0 # For huggingface_client, yalm_tokenizer, model_summac (via summarization_metrics) torchvision>=0.14.1,<3.0.0 # For huggingface_client, yalm_tokenizer, model_summac (via summarization_metrics)