Move taining scripts into the flame folder

yzhangcs · yzhangcs · commit ebb98660caf5 · 2025-03-21T21:02:32.000-07:00
diff --git a/3rdparty/flash-linear-attention b/3rdparty/flash-linear-attention
@@ -1 +1 @@
-Subproject commit 87e3cc060de50865637830ee33797d46454ca8e4
+Subproject commit a5199200efdce7eae142498f137ea77219191a21
diff --git a/flame/train.py b/flame/train.py
diff --git a/train.sh b/train.sh
@@ -102,7 +102,7 @@ torchrun --nnodes=${NNODE} \
   --local-ranks-filter ${LOG_RANK} \
   --role rank \
   --tee 3 \
-  train.py \
+  -m flame.train \
   $params
 
 echo "TRAINING DONE!"
