tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

cleanup empty lines, tweak env for runpod setup

Browse Source
This commit is contained in:
Wing Lian
2023-04-19 08:24:58 -04:00
parent f50de1b1cb
commit 7753cdee57
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@@ -152,6 +152,7 @@ use_cpu: false
- I recommend `huggingface:transformers-pytorch-deepspeed-latest-gpu` see https://hub.docker.com/r/huggingface/transformers-pytorch-deepspeed-latest-gpu/ - I recommend `huggingface:transformers-pytorch-deepspeed-latest-gpu` see https://hub.docker.com/r/huggingface/transformers-pytorch-deepspeed-latest-gpu/
- Once you start your runpod, and SSH into it: - Once you start your runpod, and SSH into it:
```shell ```shell
export TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6+PTX"
source <(curl -s https://raw.githubusercontent.com/winglian/axolotl/main/scripts/setup-runpod.sh) source <(curl -s https://raw.githubusercontent.com/winglian/axolotl/main/scripts/setup-runpod.sh)
``` ```

1
requirements.txt
View File

@@ -13,4 +13,3 @@ flash-attn
deepspeed deepspeed
einops einops
xformers xformers