fix train yaml (#82)

sotopia-lab · Mar 30, 2023 · 20dcafa · 20dcafa
1 parent 22e1ac5
commit 20dcafa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train-vicuna.yaml b/scripts/train-vicuna.yaml
@@ -65,7 +65,7 @@ run: |
   DATE=${DATE:-20230303}
   USE_FLASH_ATTN=${USE_FLASH_ATTN:-0}
   if [ $USE_FLASH_ATTN -eq 1 ]; then
-    TRAIN_SCRIPT=fastchat/train/train_flash_attn.py
+    TRAIN_SCRIPT=fastchat/train/train_mem.py
     USE_FLASH_SUFFIX="-flash"
   else
     TRAIN_SCRIPT=fastchat/train/train.py