update some logs

lightx2v/models/runners/graph_runner.py

@@ -9,10 +9,16 @@ class GraphRunner:
         self.compile()
 
     def compile(self):
-        logger.info("start compile...")
+        logger.info("=" * 60)
+        logger.info("🚀 Starting Model Compilation - Please wait, this may take a while... 🚀")
+        logger.info("=" * 60)
+
         with ProfilingContext4Debug("compile"):
             self.runner.run_step()
-        logger.info("end compile...")
+
+        logger.info("=" * 60)
+        logger.info("✅ Model Compilation Completed ✅")
+        logger.info("=" * 60)
 
     def run_pipeline(self):
         return self.runner.run_pipeline()

lightx2v/models/runners/wan/wan_audio_runner.py

@@ -360,7 +360,7 @@ class VideoGenerator:
             with ProfilingContext4Debug("step_pre"):
                 self.model.scheduler.step_pre(step_index=step_index)
 
-            with ProfilingContext4Debug("infer"):
+            with ProfilingContext4Debug("infer_main"):
                 self.model.infer(inputs)
 
             with ProfilingContext4Debug("step_post"):

lightx2v/models/runners/wan/wan_causvid_runner.py

@@ -89,7 +89,7 @@ class WanCausVidRunner(WanRunner):
                     self.model.scheduler.latents = self.model.scheduler.last_sample
                     self.model.scheduler.step_pre(step_index=self.model.scheduler.infer_steps - 1)
 
-                with ProfilingContext4Debug("infer"):
+                with ProfilingContext4Debug("infer_main"):
                     self.model.infer(self.inputs, kv_start, kv_end)
 
                 kv_start += self.num_frame_per_block * self.frame_seq_length
@@ -108,7 +108,7 @@ class WanCausVidRunner(WanRunner):
                     with ProfilingContext4Debug("step_pre"):
                         self.model.scheduler.step_pre(step_index=step_index)
 
-                    with ProfilingContext4Debug("infer"):
+                    with ProfilingContext4Debug("infer_main"):
                         self.model.infer(self.inputs, kv_start, kv_end)
 
                     with ProfilingContext4Debug("step_post"):

lightx2v/models/runners/wan/wan_skyreels_v2_df_runner.py

@@ -110,7 +110,7 @@ class WanSkyreelsV2DFRunner(WanRunner):  # Diffustion foring for SkyReelsV2 DF I
                 with ProfilingContext4Debug("step_pre"):
                     self.model.scheduler.step_pre(step_index=step_index)
 
-                with ProfilingContext4Debug("infer"):
+                with ProfilingContext4Debug("infer_main"):
                     self.model.infer(self.inputs)
 
                 with ProfilingContext4Debug("step_post"):