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