diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 9ebfe28a..deead8e7 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -35,11 +35,11 @@ dependencies: - flask-cors - Werkzeug==2.3.7 - lupa==1.10 - - transformers[sentencepiece]==4.35.0 + - transformers[sentencepiece]==4.34.0 - huggingface_hub==0.16.4 - - optimum[onnxruntime]==1.14.0 - - safetensors==0.4.0 - - accelerate==0.24.1 + - optimum[onnxruntime]==1.13.2 + - safetensors==0.3.3 + - accelerate==0.21.0 - git+https://github.com/VE-FORBRYDERNE/mkultra - flask-session - ansi2html @@ -53,9 +53,10 @@ dependencies: - git+https://github.com/0cc4m/hf_bleeding_edge/ - https://github.com/0cc4m/GPTQ-for-LLaMa/releases/download/0.0.6/gptq_koboldai-0.0.6-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' - https://github.com/0cc4m/GPTQ-for-LLaMa/releases/download/0.0.6/gptq_koboldai-0.0.6-cp38-cp38-win_amd64.whl; sys_platform == 'win32' - - auto_gptq==0.5.0 + - https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+cu118-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' + - https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+cu118-cp38-cp38-win_amd64.whl; sys_platform == 'win32' - einops - - peft==0.6.0 + - peft==0.3.0 - scipy - https://github.com/0cc4m/exllama/releases/download/0.0.7/exllama-0.0.7-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' - https://github.com/0cc4m/exllama/releases/download/0.0.7/exllama-0.0.7-cp38-cp38-win_amd64.whl; sys_platform == 'win32' diff --git a/environments/ipex.yml b/environments/ipex.yml index 0b08855c..c9794e48 100644 --- a/environments/ipex.yml +++ b/environments/ipex.yml @@ -35,11 +35,11 @@ dependencies: - flask-cors - Werkzeug==2.3.7 - lupa==1.10 - - transformers[sentencepiece]==4.35.0 + - transformers[sentencepiece]==4.34.0 - huggingface_hub==0.16.4 - - optimum[onnxruntime,openvino,nncf,neural-compressor]==1.14.0 - - safetensors==0.4.0 - - accelerate==0.24.1 + - optimum[onnxruntime,openvino,nncf,neural-compressor]==1.13.2 + - safetensors==0.3.3 + - accelerate==0.21.0 - git+https://github.com/VE-FORBRYDERNE/mkultra - flask-session - ansi2html @@ -51,9 +51,10 @@ dependencies: - git+https://github.com/0cc4m/hf_bleeding_edge/ - https://github.com/0cc4m/GPTQ-for-LLaMa/releases/download/0.0.6/gptq_koboldai-0.0.6-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' - https://github.com/0cc4m/GPTQ-for-LLaMa/releases/download/0.0.6/gptq_koboldai-0.0.6-cp38-cp38-win_amd64.whl; sys_platform == 'win32' - - auto_gptq==0.5.0 + - https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+cu118-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' + - https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+cu118-cp38-cp38-win_amd64.whl; sys_platform == 'win32' - einops - - peft==0.6.0 + - peft==0.3.0 - scipy - https://github.com/0cc4m/exllama/releases/download/0.0.7/exllama-0.0.7-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' - https://github.com/0cc4m/exllama/releases/download/0.0.7/exllama-0.0.7-cp38-cp38-win_amd64.whl; sys_platform == 'win32' diff --git a/environments/rocm.yml b/environments/rocm.yml index bb44adca..2a6043ea 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -31,11 +31,11 @@ dependencies: - flask-cors - Werkzeug==2.3.7 - lupa==1.10 - - transformers[sentencepiece]==4.35.0 + - transformers[sentencepiece]==4.34.0 - huggingface_hub==0.16.4 - - optimum[onnxruntime]==1.14.0 - - safetensors==0.4.0 - - accelerate==0.24.1 + - optimum[onnxruntime]==1.13.2 + - safetensors==0.3.3 + - accelerate==0.21.0 - git+https://github.com/VE-FORBRYDERNE/mkultra - ansi2html - flask_compress @@ -45,8 +45,8 @@ dependencies: - diffusers - git+https://github.com/0cc4m/hf_bleeding_edge/ - einops - - peft==0.6.0 + - peft==0.3.0 - windows-curses; sys_platform == 'win32' - pynvml - - auto_gptq==0.5.0 + - https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+rocm5.4.2-cp38-cp38-linux_x86_64.whl - omegaconf \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 6776effb..f668dd88 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ -transformers[sentencepiece]==4.35.0 +transformers[sentencepiece]==4.34.0 huggingface_hub==0.16.4 -optimum[onnxruntime]==1.14.0 -safetensors==0.4.0 +optimum[onnxruntime]==1.13.2 +safetensors==0.3.3 Flask==2.3.3 Flask-SocketIO==5.3.2 Werkzeug==2.3.7 @@ -17,7 +17,7 @@ lupa==1.10 markdown bleach==4.1.0 protobuf -accelerate==0.24.1 +accelerate==0.21.0 flask-session==0.5.0 marshmallow>=3.13 apispec-webframeworks @@ -40,9 +40,12 @@ pytest-metadata==2.0.4 requests-mock==1.10.0 git+https://github.com/0cc4m/hf_bleeding_edge/ einops -peft==0.6.0 +peft==0.3.0 scipy -auto-gptq==0.5.0 +https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+cu118-cp310-cp310-linux_x86_64.whl; sys_platform == 'linux' and python_version == '3.10' +https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+cu118-cp310-cp310-win_amd64.whl; sys_platform == 'win32' and python_version == '3.10' +https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+cu118-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' and python_version == '3.8' +https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+cu118-cp38-cp38-win_amd64.whl; sys_platform == 'win32' and python_version == '3.8' windows-curses; sys_platform == 'win32' pynvml https://github.com/Dao-AILab/flash-attention/releases/download/v2.3.0/flash_attn-2.3.0+cu118torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl; sys_platform == 'linux' and python_version == '3.10'