debug log

Signed-off-by: David Chen <530634352@qq.com>

Author: david6666666

vllm_omni/diffusion/registry.py

@@ -193,6 +193,12 @@ def _apply_sequence_parallel_if_enabled(model, od_config: OmniDiffusionConfig) -
 
             plan = get_sp_plan_from_model(transformer)
             if plan is None:
+                logger.info(
+                    "Sequence parallel enabled (sp_size=%s) but %s.%s has no _sp_plan.",
+                    sp_size,
+                    model.__class__.__name__,
+                    attr,
+                )
                 continue
 
             # Create SP config
@@ -211,6 +217,12 @@ def _apply_sequence_parallel_if_enabled(model, od_config: OmniDiffusionConfig) -
                 f"Applying sequence parallelism to {transformer.__class__.__name__} ({attr}) "
                 f"(sp_size={sp_size}, mode={mode}, ulysses={sp_config.ulysses_degree}, ring={sp_config.ring_degree})"
             )
+            logger.info(
+                "SP plan keys for %s.%s: %s",
+                model.__class__.__name__,
+                attr,
+                list(plan.keys()),
+            )
             apply_sequence_parallel(transformer, sp_config, plan)
             applied_count += 1