pin datasets to exact version

mosaicml · Feb 5, 2024 · 43f64fc · 43f64fc
1 parent b8dcea9
commit 43f64fc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/end-to-end-examples/stable_diffusion/requirements.txt b/examples/end-to-end-examples/stable_diffusion/requirements.txt
@@ -10,7 +10,7 @@ omegaconf==2.3.0
 # -------- diffusion -------- #
 diffusers[torch]==0.11.1
 transformers[torch]==4.25.1
-datasets>=2.14.6
+datasets==2.14.6
 pydantic==1.10.10
 scipy==1.11.1
 

diff --git a/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements-cpu.txt b/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements-cpu.txt
@@ -9,5 +9,5 @@ omegaconf
 # -------- diffusion -------- #
 diffusers[torch]==0.11.1
 transformers[torch]==4.25.1
-datasets>=2.14.6
+datasets==2.14.6
 
diff --git a/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements.txt b/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements.txt
@@ -9,7 +9,7 @@ omegaconf
 # -------- diffusion -------- #
 diffusers[torch]==0.11.1
 transformers[torch]==4.25.1
-datasets>=2.14.6
+datasets==2.14.6
 
 # -------- xformers -------- #
 xformers