fix: warn user to install mamba_ssm package (#1019)

2024-01-10 16:50:56 +09:00
parent 9e3f0cb5a7
commit d69ba2b0b7
4 changed files with 24 additions and 10 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 --extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
-packaging
+packaging==23.2
 peft==0.7.0
 transformers @ git+https://github.com/huggingface/transformers.git@3cefac1d974db5e2825a0cb2b842883a628be7a0
 tokenizers==0.15.0
@@ -34,6 +34,8 @@ fschat==0.2.34
 gradio==3.50.2
 tensorboard

+mamba-ssm==1.1.1
+
 # remote filesystems
 s3fs
 gcsfs