diff --git a/training/mochi-1/text_to_video_lora.py b/training/mochi-1/text_to_video_lora.py
index faa6255c..fb0d69a9 100644
--- a/training/mochi-1/text_to_video_lora.py
+++ b/training/mochi-1/text_to_video_lora.py
@@ -110,18 +110,16 @@ def save_model_card(
 pipe.load_lora_weights("CHANGE_ME")
 pipe.enable_model_cpu_offload()
 
-pipeline_args = {
-    "prompt": "CHANGE_ME",
-    "guidance_scale": 6.0,
-    "num_inference_steps": 64,
-    "height": 480,
-    "width": 848,
-    "max_sequence_length": 256,
-    "output_type": "np",
-}
-
-with torch.autocast("cuda", torch.bfloat16)
-    video = pipe(**pipeline_args).frames[0]
+with torch.autocast("cuda", torch.bfloat16):
+    video = pipe(
+        prompt="CHANGE_ME",
+        guidance_scale=6.0,
+        num_inference_steps=64,
+        height=480,
+        width=848,
+        max_sequence_length=256,
+        output_type="np"
+    ).frames[0]
 export_to_video(video)
 ```