fix congif (#935)

huggingface · Oct 10, 2024 · 8b574d0 · 8b574d0
1 parent 973e034
commit 8b574d0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/optimum/intel/ipex/modeling_base.py b/optimum/intel/ipex/modeling_base.py
@@ -131,6 +131,8 @@ def __init__(
         else:
             self._device = torch.device("cpu")
 
+        config = config or model.config
+
         OptimizedModel.__init__(self, model=model, config=config)
 
         self.model.to(self._device)