[deepspeed] saving checkpoint fallback when fp16 weights aren't saved (#14948)

* [deepspeed] saving checkpoint fallback when fp16 weights aren't saved * Bump required deepspeed version to match usage when saving checkpoints * update version Co-authored-by: Mihai Balint <balint.mihai@gmail.com>
2022-01-28 11:05:47 -08:00
parent d25e25ee2b
commit 297602c7f4
3 changed files with 8 additions and 3 deletions
--- a/setup.py
+++ b/setup.py
@@ -98,7 +98,7 @@ _deps = [
    "cookiecutter==1.7.2",
    "dataclasses",
    "datasets",
-    "deepspeed>=0.5.7",
+    "deepspeed>=0.5.9",
    "fairscale>0.3",
    "faiss-cpu",
    "fastapi",