17 lines
369 B
YAML
17 lines
369 B
YAML
|
name: koboldai-ft
|
||
|
|
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/finetuneanon/transformers@gpt-neo-localattention3-rp-b
|