Checking model type in nncf.torch.wrap_model (#2651)

### Changes Added model type check in nncf.torch.wrap_model ### Reason for changes Improve UX ### Related tickets #2636 (comment) ### Tests N/A
openvinotoolkit · Apr 24, 2024 · 057c4ce · 057c4ce
1 parent 9c7a212
commit 057c4ce
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/nncf/torch/model_creation.py b/nncf/torch/model_creation.py
@@ -326,6 +326,11 @@ def wrap_model(
     :param trace_parameters: Whether to trace model parameters. Default is False.
     :return: A model wrapped by NNCFNetwork.
     """
+    if not isinstance(model, torch.nn.Module):
+        raise TypeError(
+            f"The provided model type {type(model)} is incompatible. "
+            "Only models inheriting from torch.nn.Module are supported."
+        )
 
     input_info = ExampleInputInfo.from_example_input(example_input)