KoboldAI-Client/environments/rocm.yml

29 lines
500 B
YAML
Raw Normal View History

name: koboldai
channels:
- conda-forge
- defaults
dependencies:
- colorama
- flask-socketio
2022-07-04 20:07:11 +02:00
- flask-session
- python=3.8.*
- eventlet
- markdown
2022-04-12 10:32:27 +02:00
- bleach=4.1.0
- pip
2022-04-27 18:20:43 +02:00
- git=2.35.1
2022-02-21 21:17:12 +01:00
- sentencepiece
- protobuf
- marshmallow>=3.13
- apispec-webframeworks
2022-09-16 00:52:15 +02:00
- loguru
- pip:
2022-09-25 19:33:18 +02:00
- --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
- torch
2022-03-20 00:00:21 +01:00
- torchvision
- flask-cloudflared
- flask-ngrok
- lupa==1.10
- transformers>=4.20.1
- accelerate