From db0fcf4e035b3ae0cb30c5752e81ef6eb12ece54 Mon Sep 17 00:00:00 2001
From: hzg0601 <hzg0601@163.com>
Date: Fri, 14 Jul 2023 15:10:33 +0800
Subject: [PATCH] print error

---
 models/loader/loader.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/loader/loader.py b/models/loader/loader.py
index 5056932..ae4b408 100644
--- a/models/loader/loader.py
+++ b/models/loader/loader.py
@@ -454,6 +454,7 @@ class LoaderCheckPoint:
                 self.model_config.pre_seq_len = prefix_encoder_config['pre_seq_len']
                 self.model_config.prefix_projection = prefix_encoder_config['prefix_projection']
             except Exception as e:
+                print(e)
                 print("加载PrefixEncoder config.json失败")
 
         self.model, self.tokenizer = self._load_model()
@@ -471,6 +472,7 @@ class LoaderCheckPoint:
                 self.model.transformer.prefix_encoder.load_state_dict(new_prefix_state_dict)
                 self.model.transformer.prefix_encoder.float()
             except Exception as e:
+                print(e)
                 print("加载PrefixEncoder模型参数失败")
         # llama-cpp模型（至少vicuna-13b）的eval方法就是自身，其没有eval方法
         if not self.is_llamacpp: