Merge branch 'main' into perplexity-fix

nod-ai · Oct 30, 2024 · c736202 · c736202
2 parents d986174 + 560e618
commit c736202
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"))