Add safetensors support

2023-02-03 18:36:32 -03:00
parent 93b0d1b1b8
commit 03f084f311
2 changed files with 4 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,3 +4,4 @@ deepspeed==0.8.0
 gradio==3.15.0
 transformers==4.25.1
 beautifulsoup4
+safetensors