fix ruff check

Signed-off-by: jiqing-feng <[email protected]>
huggingface · Dec 11, 2024 · 24b988c · 24b988c
1 parent 2ac7ecf
commit 24b988c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/intel/ipex/modeling_base.py b/optimum/intel/ipex/modeling_base.py
@@ -193,10 +193,10 @@ def maybe_apply_torch_compile(self):
             return
         if self.use_cache and not self._supports_static_cache:
             return
-        from torch._inductor import config
+        from torch._inductor import config as inductor_config
 
         # System level optimization
-        torch._inductor.config.cpp_wrapper = True
+        inductor_config.cpp_wrapper = True
         os.environ["TORCHINDUCTOR_FREEZING"] = "1"
         logger.info("Enable torch.compile optimization")
         self.model.forward = torch.compile(self.model.forward)