diff --git a/py/libs/xyplot.py b/py/libs/xyplot.py index 2a2ff51..a3e9ce0 100644 --- 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"]: