Merge pull request ModelTC#3 from hiworldwzj/main

add dep package "safetensors" in setup.py and requirements.txt for feature "support llama model load from safetensor format"
oreo-lp · Jul 27, 2023 · bfbf80a · bfbf80a
2 parents 3c2eb35 + cdf7fd9
commit bfbf80a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -71,3 +71,4 @@ uvicorn==0.19.0
 uvloop==0.17.0
 wheel==0.40.0
 zstandard==0.19.0
+safetensors==0.3.1
diff --git a/setup.py b/setup.py
@@ -26,6 +26,7 @@
  "packaging",
  "rpyc",
  "ninja",
+ "safetensors",
  "triton==2.0.0.dev20221202"
  ],
 )