fix: pin huggingface version to <0.23.0 (#9483)

determined-ai · Jun 10, 2024 · a5cf959 · a5cf959
1 parent 97667c5
commit a5cf959
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 1 deletion.
diff --git a/examples/diffusion/textual_inversion_stable_diffusion/requirements.txt b/examples/diffusion/textual_inversion_stable_diffusion/requirements.txt
@@ -0,0 +1,6 @@
+diffusers[training]==0.5.1
+transformers==4.21.3
+ftfy==6.1.1
+setuptools==59.5.0
+accelerate==0.12.0
+huggingface_hub<0.23.0
diff --git a/examples/diffusion/textual_inversion_stable_diffusion/startup-hook.sh b/examples/diffusion/textual_inversion_stable_diffusion/startup-hook.sh
@@ -1 +1,3 @@
-pip install diffusers["training"]==0.5.1 transformers==4.21.3 ftfy==6.1.1 setuptools==59.5.0 accelerate==0.12.0
+#!/bin/bash
+
+pip install -r requirements.txt
diff --git a/examples/features/torch_batch_process_embeddings/requirements.txt b/examples/features/torch_batch_process_embeddings/requirements.txt
@@ -3,3 +3,4 @@ transformers>=4.28.1,<4.29.0
 tokenizers>=0.13.3
 datasets
 chromadb
+huggingface_hub<0.23.0
diff --git a/examples/hf_trainer_api/hf_image_classification/requirements.txt b/examples/hf_trainer_api/hf_image_classification/requirements.txt
@@ -6,3 +6,4 @@ datasets
 evaluate
 transformers>=4.28.1,<4.29.0
 scikit-learn
+huggingface_hub<0.23.0
diff --git a/examples/hf_trainer_api/hf_language_modeling/requirements.txt b/examples/hf_trainer_api/hf_language_modeling/requirements.txt
@@ -10,3 +10,4 @@ rouge_score
 fire
 sentencepiece
 scikit-learn
+huggingface_hub<0.23.0