update requirements

Oneflow-Inc · Mar 29, 2024 · 72fb26d · 72fb26d
1 parent 243f41c
commit 72fb26d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -27,3 +27,4 @@ black==21.4b2
 autoflake
 tensorboardX<=2.5.1
 pytest
+safetensors
diff --git a/setup.py b/setup.py
@@ -140,6 +140,7 @@ def get_libai_configs() -> List[str]:
             "autoflake",
             "tensorboardX<=2.5.1",
             "pytest",
+            "safetensors",
         ],
         packages=find_packages(),
         package_data={"libai.config": get_libai_configs()},