From 49a3cf132e4ad7c486d58d3e58d99a3bc6aae1b8 Mon Sep 17 00:00:00 2001 From: Henk Date: Thu, 16 Jun 2022 20:38:58 +0200 Subject: [PATCH] Require accelerate Transformers 4.20 now requires accelerate to be installed for some of the features we use in KoboldAI. This is now a required dependency for updated users. --- environments/huggingface.yml | 3 ++- environments/rocm.yml | 1 + requirements.txt | 1 + 3 files changed, 4 insertions(+), 1 deletion(-) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index da11ce10..d65f9446 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -20,4 +20,5 @@ dependencies: - flask-cloudflared - flask-ngrok - lupa==1.10 - - transformers>=4.20 \ No newline at end of file + - transformers>=4.20 + - accelerate \ No newline at end of file diff --git a/environments/rocm.yml b/environments/rocm.yml index 1fd66d26..d3e10515 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -21,3 +21,4 @@ dependencies: - flask-ngrok - lupa==1.10 - transformers>=4.20 + - accelerate diff --git a/requirements.txt b/requirements.txt index 67c6ce74..e6308790 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,3 +11,4 @@ markdown bleach==4.1.0 sentencepiece protobuf +accelerate \ No newline at end of file