Pipe in dtype correctly to model.py

Author: jackzhxng

examples/models/llama/export_llama_lib.py

@@ -983,6 +983,7 @@ def _load_llama_model(
             enable_dynamic_shape=enable_dynamic_shape,
             input_prune_map_path=input_prune_map_path,
             output_prune_map_path=output_prune_map_path,
+            dtype=dtype_override.to_torch_dtype(),
             args=args,
         )
     )

examples/models/llama/model.py

@@ -54,7 +54,7 @@ def __init__(self, **kwargs):
         self.output_prune_map_path = kwargs.get("output_prune_map_path", None)
         self.max_seq_len = kwargs.get("max_seq_len", 128)
         self.max_context_len = kwargs.get("max_context_len", 128)
-        self.dtype = kwargs.get("dtype_override", None)
+        self.dtype = kwargs.get("dtype", None)
         self.args = kwargs.get("args", None)
 
         assert (