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

ROCm Conda 

Allows anyone to easily create a ROCm compatible conda environment. Currently set to the newer transformers, you can edit the github link if you want the finetune one.
This commit is contained in:
henk717 2021-11-17 22:15:01 +01:00
parent ffdc5fc276
commit 485034b6bb
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
environments/rocm.yml Normal file
View File

@ -0,0 +1,16 @@
name: koboldai
channels:
- conda-forge
- defaults
dependencies:
- colorama
- flask-socketio
- python=3.8.*
- pip
- git
- pip:
- --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html
- torch
- torchvision==0.11.1
- flask-cloudflared
- git+https://github.com/vfbd/transformers#tfs