mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-02-09 00:08:53 +01:00
Finetune version of rocm
Seperate file so people can easily go back to the legacy implementation based on finetune (Recommended until Huggingface's compatibility is improved) . You can install and use both.
This commit is contained in:
parent
50defbaa04
commit
409be6645a
16
environments/rocm-finetune.yml
Normal file
16
environments/rocm-finetune.yml
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
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
|
Loading…
x
Reference in New Issue
Block a user