Update optimum/intel/ipex/modeling_base.py

Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>

Author: jiqing-feng

optimum/intel/ipex/modeling_base.py

@@ -194,7 +194,7 @@ def _from_transformers(
         traced_model = ipex_jit_trace(model, task, use_cache)
 
         config.torchscript = True
-        config.torch_dtype = model_kwargs.get("torch_dtype", None)
+        config.torch_dtype = torch_dtype
 
         return cls(traced_model, config=config, model_save_dir=model_id, use_cache=use_cache, warmup=False)