update dependencies to fix dataset download

modal-labs · Jan 3, 2025 · 524e9e5 · 524e9e5
1 parent 21ce643
commit 524e9e5
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/06_gpu_and_ml/openai_whisper/batched_whisper.py b/06_gpu_and_ml/openai_whisper/batched_whisper.py
@@ -29,14 +29,14 @@
 image = (
     modal.Image.debian_slim(python_version="3.11")
     .pip_install(
-        "torch",
-        "transformers",
-        "hf-transfer",
-        "huggingface_hub",
+        "torch==2.5.1",
+        "transformers==4.47.1",
+        "hf-transfer==0.1.8",
+        "huggingface_hub==0.27.0",
         "librosa==0.10.2",
         "soundfile==0.12.1",
-        "accelerate",
-        "datasets",
+        "accelerate==1.2.1",
+        "datasets==3.2.0",
     )
     # Use the barebones `hf-transfer` package for maximum download speeds. No progress bar, but expect 700MB/s.
     .env({"HF_HUB_ENABLE_HF_TRANSFER": "1"})