From 2461869aaeb62366cf6b9b08bea669e5664d6f46 Mon Sep 17 00:00:00 2001
From: yolain <yolain@163.com>
Date: Wed, 4 Dec 2024 16:38:35 +0800
Subject: [PATCH] Fix xyplot checkpoint bug #565

---
 py/libs/xyplot.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 @@ class easyXYPlot():
         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"]: