Pin Python and dep versions for embedding download (#564)

modal-labs · Jan 25, 2024 · 2025d01 · 2025d01
1 parent b4b2ad4
commit 2025d01
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/06_gpu_and_ml/embeddings/wikipedia/download.py b/06_gpu_and_ml/embeddings/wikipedia/download.py
@@ -8,7 +8,9 @@
 
 # We define our Modal Resources that we'll need
 volume = Volume.persisted("embedding-wikipedia")
-image = Image.debian_slim().pip_install("datasets", "apache_beam")
+image = Image.debian_slim(python_version="3.9").pip_install(
+    "datasets==2.16.1", "apache_beam==2.53.0"
+)
 stub = Stub(image=image)