mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-02-03 10:57:35 +01:00
485034b6bb
Allows anyone to easily create a ROCm compatible conda environment. Currently set to the newer transformers, you can edit the github link if you want the finetune one.
17 lines
333 B
YAML
17 lines
333 B
YAML
name: koboldai
|
|
channels:
|
|
- conda-forge
|
|
- defaults
|
|
dependencies:
|
|
- colorama
|
|
- flask-socketio
|
|
- python=3.8.*
|
|
- pip
|
|
- git
|
|
- pip:
|
|
- --find-links https://download.pytorch.org/whl/rocm4.2/torch_stable.html
|
|
- torch
|
|
- torchvision==0.11.1
|
|
- flask-cloudflared
|
|
- git+https://github.com/vfbd/transformers#tfs
|