Skip to content

Commit 9ab6cb8

Browse files
committed
fix export
1 parent 74541f4 commit 9ab6cb8

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

_doc/examples/plot_export_hub_codellama.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,11 @@
9393

9494
with bypass_export_some_errors(patch_transformers=True) as f:
9595
ep = torch.export.export(
96-
model, (), kwargs=f(data["inputs"]), dynamic_shapes=data["dynamic_shapes"]
96+
model,
97+
(),
98+
kwargs=f(data["inputs"]),
99+
dynamic_shapes=data["dynamic_shapes"],
100+
strict=False,
97101
)
98102
print(ep)
99103

0 commit comments

Comments
 (0)