diff --git a/sharktank/sharktank/examples/export_paged_llm_v1.py b/sharktank/sharktank/examples/export_paged_llm_v1.py index 40cfea94f..84b174bba 100644 --- a/sharktank/sharktank/examples/export_paged_llm_v1.py +++ b/sharktank/sharktank/examples/export_paged_llm_v1.py @@ -315,7 +315,7 @@ def _( print(f"EXPORT {name}:\n{ep}") print("Exporting") - output = export(fxb) + output = export(fxb, import_symbolic_shape_expressions=True) print(f"Saving to '{args.output_mlir}'") output.save_mlir(args.output_mlir) json.dump(config, open(args.output_config, "w"))