KoboldAI-Client/environments/rocm.yml

36 lines
714 B
YAML

name: koboldai
channels:
- conda-forge
- defaults
dependencies:
- colorama
- flask=2.2.3
- flask-socketio=5.3.2
- flask-session=0.4.0
- python-socketio=5.7.2
- python=3.8.*
- eventlet=0.33.3
- dnspython=2.2.1
- markdown
- bleach=4.1.0
- pip
- git=2.35.1
- sentencepiece
- protobuf
- marshmallow>=3.13
- apispec-webframeworks
- loguru
- termcolor
- psutil
- pip:
- --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
- torch==1.12.1+rocm5.1.1
- flask-cloudflared==0.0.10
- flask-ngrok
- lupa==1.10
- transformers==4.24.0
- huggingface_hub==0.12.1
- safetensors
- accelerate
- git+https://github.com/VE-FORBRYDERNE/mkultra