KoboldAI-Client/environments/rocm.yml
henk717 20e0b59fb9
Migrate to official transformers
No longer using VE's fork since its obsolete, in line with what we already did for the CUDA varient.
2021-12-07 23:35:28 +01:00

19 lines
333 B
YAML

name: koboldai
channels:
- huggingface
- conda-forge
- defaults
dependencies:
- transformers
- 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
- flask-ngrok