Merge pull request #2 from MihaiBalint/save_fp16_model-fallback

Bump required deepspeed version to match usage when saving checkpoints
stas00 · Jan 28, 2022 · 1e4cd32 · 1e4cd32
2 parents 3e749a8 + fef8ab4
commit 1e4cd32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
@@ -8,7 +8,7 @@
     "cookiecutter": "cookiecutter==1.7.2",
     "dataclasses": "dataclasses",
     "datasets": "datasets",
-    "deepspeed": "deepspeed>=0.5.7",
+    "deepspeed": "deepspeed>=0.5.9",
     "fairscale": "fairscale>0.3",
     "faiss-cpu": "faiss-cpu",
     "fastapi": "fastapi",