From 1fed1b0524ab1e578000fcf46478ee9ee708240e Mon Sep 17 00:00:00 2001 From: Henk Date: Wed, 12 Jul 2023 22:00:17 +0200 Subject: [PATCH] New dependencies --- environments/huggingface.yml | 7 ++++--- environments/rocm.yml | 4 ++-- requirements.txt | 7 ++++--- 3 files changed, 10 insertions(+), 8 deletions(-) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index fc46c402..1dab38b3 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -32,16 +32,17 @@ dependencies: - flask-ngrok - flask-cors - lupa==1.10 - - transformers==4.28.* + - transformers==4.30.1 - huggingface_hub==0.15.1 - safetensors==0.3.1 - - accelerate==0.18.0 + - accelerate==0.20.3 - git+https://github.com/VE-FORBRYDERNE/mkultra - flask-session - ansi2html - flask_compress - ijson - - bitsandbytes + - bitsandbytes==0.40.0; sys_platform == 'linux' + - https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.40.0-py3-none-win_amd64.whl; sys_platform == 'win32' - ftfy - pydub - diffusers diff --git a/environments/rocm.yml b/environments/rocm.yml index f975370f..7099474d 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -30,10 +30,10 @@ dependencies: - flask-ngrok - flask-cors - lupa==1.10 - - transformers==4.28.* + - transformers==4.30.1 - huggingface_hub==0.15.1 - safetensors==0.3.1 - - accelerate==0.18.0 + - accelerate==0.20.3 - git+https://github.com/VE-FORBRYDERNE/mkultra - ansi2html - flask_compress diff --git a/requirements.txt b/requirements.txt index 6ed3f38d..3219c08b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -transformers==4.28.* +transformers==4.30.* huggingface_hub==0.15.1 Flask==2.2.3 Flask-SocketIO==5.3.2 @@ -15,7 +15,7 @@ markdown bleach==4.1.0 sentencepiece protobuf -accelerate==0.18.0 +accelerate==0.20.3 flask-session==0.4.0 marshmallow>=3.13 apispec-webframeworks @@ -28,7 +28,8 @@ psutil ansi2html flask_compress ijson -bitsandbytes +bitsandbytes==0.40.0; sys_platform == 'linux' +https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.40.0-py3-none-win_amd64.whl; sys_platform == 'win32' ftfy py==1.11.0 pydub