diff --git a/examples/models/llama2/export_llama_lib.py b/examples/models/llama2/export_llama_lib.py index 5156189adca..9f8569c2865 100644 --- a/examples/models/llama2/export_llama_lib.py +++ b/examples/models/llama2/export_llama_lib.py @@ -374,8 +374,8 @@ def _prepare_for_llama_export(modelname: str, args) -> LlamaEdgeManager: ) .set_output_dir(output_dir_path) .set_metadata(args.metadata) - .source_transform(transforms) .to_dtype(dtype_override) + .source_transform(transforms) )