diff --git a/debug/accuracy_tools/msprobe/core/service.py b/debug/accuracy_tools/msprobe/core/service.py index af31a15d3dec41bbc960e19db92c3d6267295109..a914c23031c38859bbdb770826e269ef2cedf372 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)