diff --git a/training/mochi-1/text_to_video_lora.py b/training/mochi-1/text_to_video_lora.py
index dc4f30a..faa6255 100644
--- a/training/mochi-1/text_to_video_lora.py
+++ b/training/mochi-1/text_to_video_lora.py
@@ -89,9 +89,9 @@ def save_model_card(
 
 ## Model description
 
-This is a lora finetune of the Moch-1 preview model `{base_model}`.
+This is a lora finetune of the Mochi-1 preview model `{base_model}`.
 
-The model was trained using [CogVideoX Factory](https://github.com/a-r-r-o-w/cogvideox-factory) - a repository containing memory-optimized training scripts for the CogVideoX, Mochi family of models using [TorchAO](https://github.com/pytorch/ao) and [DeepSpeed](https://github.com/microsoft/DeepSpeed). The scripts were adopted from [CogVideoX Diffusers trainer](https://github.com/huggingface/diffusers/blob/main/examples/cogvideo/train_cogvideox_lora.py).
+The model was trained using [CogVideoX Factory](https://github.com/a-r-r-o-w/cogvideox-factory) - a repository containing memory-optimized training scripts for the CogVideoX and Mochi family of models using [TorchAO](https://github.com/pytorch/ao) and [DeepSpeed](https://github.com/microsoft/DeepSpeed). The scripts were adopted from [CogVideoX Diffusers trainer](https://github.com/huggingface/diffusers/blob/main/examples/cogvideo/train_cogvideox_lora.py).
 
 ## Download model
 
diff --git a/training/mochi-1/train.sh b/training/mochi-1/train.sh
index f789281..2c378e2 100644
--- a/training/mochi-1/train.sh
+++ b/training/mochi-1/train.sh
@@ -8,7 +8,7 @@ DATA_ROOT="videos_prepared"
 MODEL="genmo/mochi-1-preview"
 OUTPUT_PATH="mochi-lora"
 
-cmd="CUDA_VISIBLE_DEVICES=$GPU_IDS python text_to_video_lora_simple.py \
+cmd="CUDA_VISIBLE_DEVICES=$GPU_IDS python text_to_video_lora.py \
   --pretrained_model_name_or_path $MODEL \
   --cast_dit \
   --data_root $DATA_ROOT \