style: fix new ruff error

huggingface · Jun 28, 2024 · 0eba37f · 0eba37f
1 parent 05791bd
commit 0eba37f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/quanto/nn/qmodule.py b/optimum/quanto/nn/qmodule.py
@@ -173,7 +173,7 @@ def _load_from_state_dict(
             if assign_to_params_buffers:
                 self.weight = torch.nn.Parameter(deserialized_weight)
             else:
-                if type(self.weight.data) != type(deserialized_weight):
+                if type(self.weight.data) is not type(deserialized_weight):
                     # Reloading frozen weights into unfrozen module: move to the correct device and force assignment
                     self.weight = torch.nn.Parameter(deserialized_weight.to(self.weight.device))
                 else: