diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py
index a16f8067d4..bcd1996b3b 100644
--- a/src/autotrain/__init__.py
+++ b/src/autotrain/__init__.py
@@ -41,4 +41,4 @@
 
 
 logger = Logger().get_logger()
-__version__ = "0.7.55.dev0"
+__version__ = "0.7.56.dev0"
diff --git a/src/autotrain/cli/run_llm.py b/src/autotrain/cli/run_llm.py
index 84fbd432ee..b6589762f7 100644
--- a/src/autotrain/cli/run_llm.py
+++ b/src/autotrain/cli/run_llm.py
@@ -191,7 +191,23 @@ def register_subcommand(parser: ArgumentParser):
                 "required": False,
                 "type": int,
                 "default": 1024,
-                "alias": ["--max-len", "--max-length"],
+                "alias": ["--model-max-length"],
+            },
+            {
+                "arg": "--max_prompt_length",
+                "help": "Prompt length to use, for orpo",
+                "required": False,
+                "type": int,
+                "default": 128,
+                "alias": ["--max-prompt-length"],
+            },
+            {
+                "arg": "--max_completion_length",
+                "help": "Completion length to use, for orpo: encoder-decoder models only",
+                "required": False,
+                "type": int,
+                "default": None,
+                "alias": ["--max-completion-length"],
             },
             {
                 "arg": "--trainer",