Merge branch 'perplexity-fix' of https://github.com/nod-ai/SHARK-Plat…

…form into perplexity-fix
nod-ai · Oct 31, 2024 · a505de5 · a505de5
2 parents 708c855 + c736202
commit a505de5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"))