Update nodes.py

2025-10-30 17:33:42 +02:00
parent cc9bf1e4f5
commit 64c195167b
1 changed files with 3 additions and 1 deletions
@@ -1860,7 +1860,9 @@ class WanVideoScheduler: #WIP
                if start_idx > 0 and 0 <= start_idx < len(sigmas_np):
                    ax.axvline(start_idx, color='green', linestyle='--', linewidth=2, label='start_step split')
                if (end_idx != -1 and 0 <= end_idx < len(sigmas_np)) or (start_idx > 0 and 0 <= start_idx < len(sigmas_np)):
-                    ax.legend()
+                    handles, labels = ax.get_legend_handles_labels()
+                    if labels:
+                        ax.legend()
                if start_idx < end_idx and 0 <= start_idx < len(sigmas_np) and 0 < end_idx < len(sigmas_np):
                    ax.axvspan(start_idx, end_idx, color='lightblue', alpha=0.1, label='Sampled Range')
                plt.tight_layout()