KoboldAI-Client/environments/huggingface.yml
henk717 fcc210898f Revert to Python 3.8
tensorflow-base doesn't like the older cudatoolkit anymore. The one for python 3.8 still does, so lets just stick to that for now then.
2021-09-16 01:18:04 +02:00

17 lines
254 B
YAML

name: koboldai
channels:
- huggingface
- pytorch
- conda-forge
- defaults
dependencies:
- colorama
- flask-socketio
- pytorch
- python=3.8.*
- cudatoolkit=11.1
- tensorflow-gpu
- transformers
- pip
- pip:
- flask-cloudflared