From 89637ae9d7226cbc039af571fbefc6e4feb31b66 Mon Sep 17 00:00:00 2001 From: Henk Date: Sun, 23 Jul 2023 22:51:47 +0200 Subject: [PATCH] GPTQ Requirements --- environments/huggingface.yml | 4 ++-- requirements.txt | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index e97f3e2e..89cd4c89 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -47,8 +47,8 @@ dependencies: - pydub - diffusers - git+https://github.com/0cc4m/hf_bleeding_edge/ - - --find-links=https://0cc4m.github.io/GPTQ-for-LLaMa/gptq-whl-links.html - - gptq_koboldai==0.0.6 + - 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' - einops - peft==0.3.0 - scipy diff --git a/requirements.txt b/requirements.txt index 3978eda3..c3487bb9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -39,7 +39,8 @@ pytest-metadata==2.0.4 requests-mock==1.10.0 safetensors==0.3.1 git+https://github.com/0cc4m/hf_bleeding_edge/ ---find-links=https://0cc4m.github.io/KoboldAI/gptq-whl-links.html gptq_koboldai==0.0.4 +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' einops peft==0.3.0 scipy \ No newline at end of file