rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

ROCm dependency update

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
environments/rocm.yml
View File

@ -24,18 +24,18 @@ dependencies:
- psutil
- ffmpeg
- pip:
- --extra-index-url https://download.pytorch.org/whl/rocm5.2
- torch==1.13.1+rocm5.2
- --extra-index-url https://download.pytorch.org/whl/rocm5.6
- torch==2.1.2+rocm5.6
- flask-cloudflared==0.0.10
- flask-ngrok
- flask-cors
- Werkzeug==2.3.7
- lupa==1.10
- transformers[sentencepiece]==4.34.0
- huggingface_hub==0.16.4
- optimum[onnxruntime]==1.13.2
- safetensors==0.3.3
- accelerate==0.21.0
- transformers[sentencepiece]==4.36.1
- huggingface_hub==0.19.4
- optimum[onnxruntime]==1.16.1
- safetensors==0.4.1
- accelerate==0.25.0
- git+https://github.com/VE-FORBRYDERNE/mkultra
- ansi2html
- flask_compress
@ -45,8 +45,9 @@ dependencies:
- diffusers
- git+https://github.com/0cc4m/hf_bleeding_edge/
- einops
- peft==0.3.0
- peft==0.7.1
- windows-curses; sys_platform == 'win32'
- 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
- omegaconf
- 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
- https://github.com/turboderp/exllamav2/releases/download/v0.0.11/exllamav2-0.0.11-cp38-cp38-linux_x86_64.whl