rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

ROCm dependency update

Browse Source
This commit is contained in:
Henk
2023-12-19 00:13:09 +01:00
parent e6afb766af
commit 177e551da6
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
environments/rocm.yml
View File

@@ -24,18 +24,18 @@ dependencies:
- psutil - psutil
- ffmpeg - ffmpeg
- pip: - pip:
- --extra-index-url https://download.pytorch.org/whl/rocm5.2 - --extra-index-url https://download.pytorch.org/whl/rocm5.6
- torch==1.13.1+rocm5.2 - torch==2.1.2+rocm5.6
- flask-cloudflared==0.0.10 - flask-cloudflared==0.0.10
- flask-ngrok - flask-ngrok
- flask-cors - flask-cors
- Werkzeug==2.3.7 - Werkzeug==2.3.7
- lupa==1.10 - lupa==1.10
- transformers[sentencepiece]==4.34.0 - transformers[sentencepiece]==4.36.1
- huggingface_hub==0.16.4 - huggingface_hub==0.19.4
- optimum[onnxruntime]==1.13.2 - optimum[onnxruntime]==1.16.1
- safetensors==0.3.3 - safetensors==0.4.1
- accelerate==0.21.0 - accelerate==0.25.0
- git+https://github.com/VE-FORBRYDERNE/mkultra - git+https://github.com/VE-FORBRYDERNE/mkultra
- ansi2html - ansi2html
- flask_compress - flask_compress
@@ -45,8 +45,9 @@ dependencies:
- diffusers - diffusers
- git+https://github.com/0cc4m/hf_bleeding_edge/ - git+https://github.com/0cc4m/hf_bleeding_edge/
- einops - einops
- peft==0.3.0 - peft==0.7.1
- windows-curses; sys_platform == 'win32' - windows-curses; sys_platform == 'win32'
- pynvml - pynvml
- https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.4.2/auto_gptq-0.4.2+rocm5.4.2-cp38-cp38-linux_x86_64.whl - https://huggingface.github.io/autogptq-index/whl/rocm561/auto-gptq/auto_gptq-0.5.1%2Brocm5.6.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
- omegaconf - omegaconf
- https://github.com/turboderp/exllamav2/releases/download/v0.0.11/exllamav2-0.0.11-cp38-cp38-linux_x86_64.whl