Widen PyTorch dependency version to allow PyTorch 2 (stanford-crfm#1759)

danielz02 · Sep 7, 2023 · b38e978 · b38e978
1 parent 1c50dd9
commit b38e978
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -79,8 +79,8 @@ install_requires=
     # Models and Metrics Extras
     transformers~=4.28.1  # For anthropic_client, huggingface_client, huggingface_tokenizer, test_openai_token_cost_estimator, model_summac (via summarization_metrics)
     # TODO: Upgrade torch
-    torch~=1.12.1  # For huggingface_client, yalm_tokenizer, model_summac (via summarization_metrics)
-    torchvision~=0.13.1  # For huggingface_client, yalm_tokenizer, model_summac (via summarization_metrics)
+    torch>=1.12.1,<3.0.0  # For huggingface_client, yalm_tokenizer, model_summac (via summarization_metrics)
+    torchvision>=0.13.1,<3.0.0  # For huggingface_client, yalm_tokenizer, model_summac (via summarization_metrics)
 
     # Metrics Extras
     google-api-python-client~=2.64.0  # For perspective_api_client via toxicity_metrics