KoboldAI-Client/environments/huggingface.yml
Henk 7fc5c46c1d Add Safetensors
Having the dependency adds basic support for safetensor models.
2023-01-06 16:40:56 +01:00

34 lines
580 B
YAML

name: koboldai
channels:
- pytorch
- conda-forge
- defaults
dependencies:
- colorama
- flask-socketio
- flask-session
- pytorch=1.11.*
- python=3.8.*
- cudatoolkit=11.1
- eventlet
- markdown
- bleach=4.1.0
- pip
- git=2.35.1
- sentencepiece
- protobuf
- marshmallow>=3.13
- apispec-webframeworks
- loguru
- termcolor
- psutil
- pip:
- flask-cloudflared
- flask-ngrok
- lupa==1.10
- transformers==4.24.0
- huggingface_hub>=0.10.1
- safetensors
- accelerate
- git+https://github.com/VE-FORBRYDERNE/mkultra