From 409be6645ab09e0612687521e76189a4cf4b986c Mon Sep 17 00:00:00 2001 From: henk717 Date: Sat, 20 Nov 2021 03:14:18 +0100 Subject: [PATCH] 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. --- environments/rocm-finetune.yml | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 environments/rocm-finetune.yml diff --git a/environments/rocm-finetune.yml b/environments/rocm-finetune.yml new file mode 100644 index 00000000..477dd6aa --- /dev/null +++ b/environments/rocm-finetune.yml @@ -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