cleanup empty lines, tweak env for runpod setup

2023-04-19 08:24:58 -04:00
parent f50de1b1cb
commit 7753cdee57
2 changed files with 1 additions and 1 deletions
--- a/README.md
+++ b/README.md
@@ -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/
 - Once you start your runpod, and SSH into it:
 ```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)
 ```

--- a/requirements.txt
+++ b/requirements.txt
@@ -13,4 +13,3 @@ flash-attn
 deepspeed
 einops
 xformers
-