Automated PR - 2026-03-05

packages/ltx-pipelines/src/ltx_pipelines/utils/args.py

@@ -8,6 +8,7 @@ from ltx_pipelines.utils.constants import (
     DEFAULT_IMAGE_CRF,
     DEFAULT_LORA_STRENGTH,
     DEFAULT_NEGATIVE_PROMPT,
+    LTX_2_3_HQ_PARAMS,
     LTX_2_3_PARAMS,
     PipelineParams,
 )
@@ -457,6 +458,23 @@ def default_2_stage_arg_parser(params: PipelineParams = LTX_2_3_PARAMS) -> argpa
     return parser
 
 
+def hq_2_stage_arg_parser(params: PipelineParams = LTX_2_3_HQ_PARAMS) -> argparse.ArgumentParser:
+    parser = default_2_stage_arg_parser(params=params)
+    parser.add_argument(
+        "--distilled-lora-strength-stage-1",
+        type=float,
+        default=0.25,
+        help=(f"Strength of the distilled LoRA used in the first stage (default: {0.25})."),
+    )
+    parser.add_argument(
+        "--distilled-lora-strength-stage-2",
+        type=float,
+        default=0.5,
+        help=(f"Strength of the distilled LoRA used in the second stage (default: {0.5})."),
+    )
+    return parser
+
+
 def default_2_stage_distilled_arg_parser(params: PipelineParams = LTX_2_3_PARAMS) -> argparse.ArgumentParser:
     parser = basic_arg_parser(params=params, distilled=True)
     parser.set_defaults(height=params.stage_2_height, width=params.stage_2_width)