fix spelling error

huggingface · May 31, 2024 · 4a81ea9 · 4a81ea9
1 parent bde814e
commit 4a81ea9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/intel/ipex/modeling_base.py b/optimum/intel/ipex/modeling_base.py
@@ -95,7 +95,7 @@ def ipex_jit_trace(model, task, use_cache):
 
     model.config.return_dict = False
 
-    model.config.use_cache = past_key_values" in sample_inputs 
+    model.config.use_cache = "past_key_values" in sample_inputs 
 
     model = ipex.optimize(model.eval(), dtype=model.dtype, inplace=True)
     # Disable repack while jit tracing to reduce the memory