From 24ac90e8db9ed87f93e69327dcc79d202709ec08 Mon Sep 17 00:00:00 2001 From: Yang An <yangapku@gmail.com> Date: Sat, 5 Aug 2023 23:49:15 +0800 Subject: [PATCH] Update requirements.txt --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 466f76f..5721696 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,4 +2,5 @@ transformers==4.31.0 accelerate tiktoken einops -transformers_stream_generator==0.0.4 \ No newline at end of file +transformers_stream_generator==0.0.4 +bitsandbytes