Fix xyplot checkpoint bug #565

yolain · Dec 4, 2024 · 2461869 · 2461869
1 parent bb9dc79
commit 2461869
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/libs/xyplot.py b/py/libs/xyplot.py
@@ -62,7 +62,7 @@ def define_variable(plot_image_vars, value_type, value, index):
         if value_type in ['Lora', 'Checkpoint']:
             arr = value.split(',')
             model_name = os.path.basename(os.path.splitext(arr[0])[0])
-            trigger_words = ' ' + arr[3] if len(arr[3]) > 2 else ''
+            trigger_words = ' ' + arr[3] if value_type == 'Lora' and len(arr[3]) > 2 else ''
             value_label = f"{model_name}{trigger_words}"
 
         if value_type in ["ModelMergeBlocks"]: