From 72e2798141a4dffb78f6028e28b50c33808b0fdf Mon Sep 17 00:00:00 2001
From: jiangchao_j <chaojiang_j@163.com>
Date: Mon, 11 Aug 2025 10:12:14 +0800
Subject: [PATCH] check if the model is None

---
 debug/accuracy_tools/msprobe/core/service.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/debug/accuracy_tools/msprobe/core/service.py b/debug/accuracy_tools/msprobe/core/service.py
index af31a15d3..a914c2303 100644
--- a/debug/accuracy_tools/msprobe/core/service.py
+++ b/debug/accuracy_tools/msprobe/core/service.py
@@ -119,7 +119,8 @@ class BaseService(ABC):
         self._process_iteration()
         if self._is_debug_level:
             return
-        self.model = model
+        if model:
+            self.model = model
         if self._is_need_module_hook and self.model not in self.hooked_modules:
             self._register_module_hook()
             self.hooked_modules.append(self.model)
-- 
Gitee