Pinning datasets to work around weird issue

allenai · Oct 6, 2024 · 13123dd · 13123dd
1 parent 568dd48
commit 13123dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gantry-requirements.txt b/gantry-requirements.txt
@@ -26,7 +26,7 @@ sphinx-autodoc-typehints==1.23.3
 packaging
 necessary
 accelerate>=0.34.2
-datasets>=3.0.0
+datasets==3.0.0
 peft
 wandb
 omegaconf