From e3071eb677e4827475b88771c731588908596cb5 Mon Sep 17 00:00:00 2001 From: hyperzlib Date: Sat, 27 Apr 2024 19:58:08 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0requirements?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .gitignore | 1 + requirements.txt | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/.gitignore b/.gitignore index 9a7fbea..03677b0 100644 --- a/.gitignore +++ b/.gitignore @@ -7,6 +7,7 @@ build .vscode/ .idea/ .DS_Store +.venv /private/ output_qwen/ diff --git a/requirements.txt b/requirements.txt index 9c35591..c73afa7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,3 +4,8 @@ tiktoken einops transformers_stream_generator==0.0.4 scipy +auto-gptq>=0.5.1 +transformers>=4.35.0 +optimum>=1.14.0 +peft<0.8.0 +deepspeed