KoboldAI-Client/environments/rocm.yml

28 lines
508 B
YAML

name: koboldai
channels:
- conda-forge
- defaults
dependencies:
- colorama
- flask-socketio
- flask-session
- python=3.8.*
- eventlet
- markdown
- bleach=4.1.0
- pip
- git=2.35.1
- sentencepiece
- protobuf
- marshmallow>=3.13
- apispec-webframeworks
- pip:
- --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html
- torch==1.10.*
- torchvision
- flask-cloudflared
- flask-ngrok
- lupa==1.10
- transformers>=4.20.1
- accelerate