mindstudio-probe 8.3.2__py3-none-any.whl → 26.0.0a1__py3-none-any.whl

msprobe/pytorch/compare/pt_diff_analyze.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2025-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 
 from msprobe.pytorch.compare.distributed_compare import compare_distributed

msprobe/pytorch/compare/utils.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2025-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import os
 

msprobe/pytorch/{hook_module → dump/api_dump}/api_register.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2025-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import functools
 import inspect
@@ -23,7 +24,7 @@ import torch.distributed as dist
 from msprobe.core.common.const import Const
 from msprobe.core.common.file_utils import load_yaml
 from msprobe.core.common.runtime import Runtime
-from msprobe.core.data_dump.api_registry import ApiRegistry
+from msprobe.core.dump.api_dump.api_registry import ApiRegistry
 from msprobe.pytorch.common.log import logger
 from msprobe.pytorch.common.utils import (
     torch_without_guard_version,
@@ -31,9 +32,9 @@ from msprobe.pytorch.common.utils import (
     torch_device_guard,
     parameter_adapter
 )
-from msprobe.pytorch.function_factory import npu_custom_functions
-from msprobe.pytorch.hook_module.hook_module import HOOKModule
-from msprobe.pytorch.hook_module.utils import dynamic_import_op
+from msprobe.pytorch.dump.function_factory import npu_custom_functions
+from msprobe.pytorch.dump.api_dump.hook_module import HOOKModule
+from msprobe.pytorch.dump.api_dump.utils import dynamic_import_op
 
 try:
     import mindspeed.ops
@@ -55,7 +56,8 @@ _api_types = {
         Const.PT_API_TYPE_TENSOR: ((torch.Tensor,), (torch.Tensor,)),
         Const.PT_API_TYPE_TORCH: ((torch,), (torch,)),
         Const.PT_API_TYPE_VF: ((torch._C._VariableFunctionsClass,), (torch._VF,)),
-        Const.PT_API_TYPE_DIST: ((dist,), (dist, dist.distributed_c10d))
+        Const.PT_API_TYPE_DIST: ((dist,), (dist,)),
+        Const.PT_API_TYPE_DIST_C10D: ((dist.distributed_c10d,), (dist.distributed_c10d,))
     }
 }
 if not is_gpu:

msprobe/pytorch/{hook_module → dump/api_dump}/hook_module.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2024-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import functools
 from collections import defaultdict
@@ -20,7 +21,7 @@ import torch
 import torch.nn as nn
 import torch.utils.hooks as full_hooks
 
-from msprobe.pytorch.common.utils import register_forward_pre_hook
+from msprobe.pytorch.common.utils import register_forward_pre_hook, is_float8_tensor
 
 
 class HOOKModule(nn.Module):
@@ -84,7 +85,7 @@ class HOOKModule(nn.Module):
                 else:
                     return result
 
-            if not (var.requires_grad and torch.is_grad_enabled()):
+            if is_float8_tensor(var) or not (var.requires_grad and torch.is_grad_enabled()):
                 return result
 
             grad_fn = var.grad_fn

msprobe/pytorch/{hook_module → dump/api_dump}/pt_hook_manager.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import functools
 import threading
@@ -20,19 +21,20 @@ from contextlib import nullcontext
 import torch
 
 from msprobe.core.common.const import Const
+from msprobe.core.common.log import logger
 from msprobe.core.common.runtime import Runtime
 from msprobe.core.common.utils import replace_last_occurrence, ThreadSafe
-from msprobe.core.data_dump.data_processor.base import (ModuleForwardInputsOutputs)
-from msprobe.core.hook_manager import BaseHookManager, HookSet
-from msprobe.pytorch.common.utils import is_recomputation, torch_version_above_or_equal_2, register_forward_hook
-from msprobe.pytorch.hook_module.hook_module import HOOKModule
-
+from msprobe.core.dump.data_dump.data_processor.base import ModuleForwardInputsOutputs
+from msprobe.core.dump.hook_manager import BaseHookManager, HookSet
+from msprobe.pytorch.common.utils import (
+    torch_version_above_or_equal_2,
+    register_forward_hook,
+    Const as PtConst
+)
+from msprobe.pytorch.dump.api_dump.hook_module import HOOKModule
+from msprobe.pytorch.dump.module_dump.module_processor import ModuleProcessor
 
 class PytorchHookManager(BaseHookManager):
-    @property
-    def _is_recompute(self):
-        return is_recomputation()
-
     @staticmethod
     def _no_grad_context():
         return nullcontext()
@@ -77,7 +79,15 @@ class PytorchHookManager(BaseHookManager):
     def _register_backward_hook(self, module, full_backward_name, args):
         pass
 
-    def _register_backward_pre_hook(self, module, full_backward_name, output):
+    def _register_backward_pre_hook(self, module, full_backward_name, args, kwargs, output):
+        if module.prefix_api_name in PtConst.DROPOUT_API_LIST:
+            p = kwargs.get('p', 0.5)
+            if len(args) >= 2:
+                p = args[1]
+            if p == 0:
+                logger.debug(f"The parameter 'p' is set to 0, so the {full_backward_name} data will not be collected.")
+                return output
+
         var = output
         while not isinstance(var, torch.Tensor):
             if isinstance(var, dict):
@@ -129,9 +139,44 @@ class PytorchHookManager(BaseHookManager):
                         full_name,
                         module,
                         self._pid,
-                        module_input_output,
-                        self._is_recompute
+                        module_input_output
                     )
                 BaseHookManager.inner_switch[tid] = False
 
         return distributed_forward_hook
+
+    def _register_param_hook(self, name, module, params_dict):
+        ori_name = name.rsplit(Const.SEP, 2)[0]
+        # data_mode为forward时，不注册参数hook
+        if not (Const.FORWARD in self.config.data_mode and Const.BACKWARD not in self.config.data_mode):
+            for param_name, param in params_dict.items():
+                if param.requires_grad:
+                    name = ori_name + Const.SEP + param_name
+                    old_handle = BaseHookManager.hook_handle_dict.get(name)
+                    if old_handle and hasattr(old_handle, "remove"):
+                        old_handle.remove()
+                    param_tmp = param.expand_as(param)
+                    if param_tmp.grad_fn:
+                        grad_acc = param_tmp.grad_fn.next_functions[0][0]
+                        handle = grad_acc.register_hook(self._build_grad_hook(ori_name, param_name, param))
+                        BaseHookManager.hook_handle_dict[name] = handle
+
+    def _build_grad_hook(self, ori_name, param_name, param):
+        def hook_fn(*args):
+            tid = threading.get_ident()
+            if not self._should_execute_hook(Const.MODULE, tid):
+                return
+            with ThreadSafe():
+                original_state = self.ensure_gc_enabled()
+                BaseHookManager.inner_switch[tid] = True
+                index = self._get_grad_hook_call_index(ori_name, param_name)
+                if ModuleProcessor.is_megatron_module and hasattr(param, "main_grad"):
+                    grad = param.main_grad
+                else:
+                    grad = param.grad
+                self.data_collector.params_data_collect(ori_name, param_name, self._pid, grad, str(index))
+                BaseHookManager.inner_switch[tid] = False
+                self.restore_gc_state(original_state)
+            return
+
+        return hook_fn

msprobe/pytorch/{hook_module → dump/api_dump}/register_optimizer_hook.py

@@ -1,19 +1,21 @@
-# Copyright (c) 2024-2024, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import torch
+
 from msprobe.core.common.const import Const
 from msprobe.pytorch.common.log import logger
 

msprobe/pytorch/{hook_module → dump/api_dump}/script_wrapper.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2025-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import functools
 import importlib
@@ -21,7 +22,7 @@ import torch
 
 from msprobe.core.common.log import logger
 from msprobe.pytorch.common.utils import torch_version_above_or_equal_2
-from msprobe.pytorch.hook_module.api_register import get_api_register
+from msprobe.pytorch.dump.api_dump.api_register import get_api_register
 
 if torch_version_above_or_equal_2:
     from torch._dynamo.convert_frame import convert_frame as _orig_convert_frame, Hooks
@@ -67,8 +68,10 @@ def wrap_compile_script_func():
 
         # 保留原属性以兼容
         _wrapped._torchdynamo_orig_callable = compiler_fn  # type: ignore[attr-defined]
-        _wrapped._clone_with_backend = lambda backend: _patched_convert_frame(backend,
-                                                                              hooks)  # type: ignore[attr-defined]
+
+        _wrapped._clone_with_backend =\
+            lambda backend: _patched_convert_frame(backend, hooks)  # type: ignore[attr-defined]
+
         return _wrapped
 
     import torch._dynamo.convert_frame as _cf_mod

msprobe/pytorch/{hook_module → dump/api_dump}/utils.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2024-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 import os
 import importlib

msprobe/pytorch/{debugger → dump/debugger}/debugger_config.py

@@ -1,17 +1,18 @@
-# Copyright (c) 2024-2025, Huawei Technologies Co., Ltd.
-# All rights reserved.
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 #
-# Licensed under the Apache License, Version 2.0  (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
 #
-# http://www.apache.org/licenses/LICENSE-2.0
+#          http://license.coscl.org.cn/MulanPSL2
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
 
 from msprobe.core.common.const import Const
 from msprobe.core.common.exceptions import MsprobeException
@@ -26,29 +27,15 @@ class DebuggerConfig:
         self.rank = common_config.rank if common_config.rank else []
         self.step = common_config.step if common_config.step else []
         self.level = level or common_config.level or Const.LEVEL_L1
-        self.enable_dataloader = common_config.enable_dataloader
         self.scope = task_config.scope if task_config.scope else []
         self.list = task_config.list if task_config.list else []
         self.data_mode = task_config.data_mode if task_config.data_mode else ["all"]
         self.summary_mode = task_config.summary_mode if task_config.summary_mode else Const.STATISTICS
-        self.overflow_nums = task_config.overflow_nums if task_config.overflow_nums else 1
         self.framework = Const.PT_FRAMEWORK
         self.async_dump = common_config.async_dump if common_config.async_dump else False
         self.precision = common_config.precision if common_config.precision else Const.DUMP_PRECISION_LOW
-
-        if self.task == Const.FREE_BENCHMARK:
-            self.fuzz_device = task_config.fuzz_device
-            self.handler_type = task_config.handler_type
-            self.pert_mode = task_config.pert_mode
-            self.fuzz_level = task_config.fuzz_level
-            self.fuzz_stage = task_config.fuzz_stage
-            self.preheat_config = {
-                "if_preheat": task_config.if_preheat,
-                "preheat_step": task_config.preheat_step,
-                "max_sample": task_config.max_sample
-            }
-
-
+        self.diff_nums = task_config.diff_nums if task_config.diff_nums else 1
+        self.bench_path = getattr(task_config, "bench_path", None)
         self.check()
         self._check_statistics_config(task_config)
 
@@ -57,9 +44,9 @@ class DebuggerConfig:
             self._check_and_adjust_config_with_l2()
 
     def check(self):
-        if self.task and self.task not in Const.TASK_LIST:
+        if self.task and self.task not in Const.TORCH_TASK_LIST:
             raise MsprobeException(MsprobeException.INVALID_PARAM_ERROR,
-                                   f"The task <{self.task}> is not in the {Const.TASK_LIST}.")
+                                   f"The task <{self.task}> is not in the {Const.TORCH_TASK_LIST}.")
         if self.level and self.level not in Const.LEVEL_LIST:
             raise MsprobeException(MsprobeException.INVALID_PARAM_ERROR,
                                    f"The level <{self.level}> is not in the {Const.LEVEL_LIST}.")
@@ -91,28 +78,26 @@ class DebuggerConfig:
                 )
         return True
 
-    def check_model(self, instance, start_model, token_range=None):
-        instance.model = start_model if start_model is not None else instance.model
-
-        if token_range and not instance.model:
+    def check_model(self, models, token_range=None):
+        if token_range and not models:
             error_info = "The 'model' parameter must be provided when token_range is not None"
             raise MsprobeException(MsprobeException.INVALID_PARAM_ERROR, error_info)
 
         if self.level not in [Const.LEVEL_L0, Const.LEVEL_MIX] and token_range is None:
             return
 
-        if instance.model is None:
+        if models is None:
             logger.error_on_rank_0(
                 f"For level {self.level} or non-empty token_range, "
                 f"PrecisionDebugger or start interface must receive a 'model' parameter.")
             raise MsprobeException(MsprobeException.INVALID_PARAM_ERROR, f"missing the parameter 'model'")
 
-        if is_torch_nn_module(instance.model):
+        if is_torch_nn_module(models):
             return
 
-        if isinstance(instance.model, (list, tuple)):
+        if isinstance(models, (list, tuple)):
             error_model = None
-            for model in instance.model:
+            for model in models:
                 if not is_torch_nn_module(model):
                     error_model = model
                     break
@@ -123,7 +108,7 @@ class DebuggerConfig:
                     MsprobeException.INVALID_PARAM_ERROR, error_info)
         else:
             error_info = (f"The 'model' parameter must be a torch.nn.Module or list[torch.nn.Module] "
-                          f"type, currently there is an unsupported {type(instance.model)} type.")
+                          f"type, currently there is an unsupported {type(models)} type.")
             raise MsprobeException(
                 MsprobeException.INVALID_PARAM_ERROR, error_info)