From 9c86706cbb5d66293a397f8abc1bac1225cfe589 Mon Sep 17 00:00:00 2001
From: sayakpaul <spsayakpaul@gmail.com>
Date: Fri, 29 Nov 2024 13:08:55 +0530
Subject: [PATCH] fix help message

---
 training/mochi-1/args.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/training/mochi-1/args.py b/training/mochi-1/args.py
index 4f1cc7e..51671c0 100644
--- a/training/mochi-1/args.py
+++ b/training/mochi-1/args.py
@@ -42,7 +42,7 @@ def _get_model_args(parser: argparse.ArgumentParser) -> None:
     parser.add_argument(
         "--compile_dit",
         action="store_true",
-        help="If we should cast DiT params to a lower precision.",
+        help="If we should compile the DiT.",
     )