From 2025d0156ce61d398e9130aedd71cf6c71496ba0 Mon Sep 17 00:00:00 2001 From: Akshat Bubna Date: Thu, 25 Jan 2024 14:15:10 -0800 Subject: [PATCH] Pin Python and dep versions for embedding download (#564) --- 06_gpu_and_ml/embeddings/wikipedia/download.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/06_gpu_and_ml/embeddings/wikipedia/download.py b/06_gpu_and_ml/embeddings/wikipedia/download.py index 6d0ff4990..63463f9ef 100644 --- 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)