Skip to content

Commit cdd0be5

Browse files
authored
Merge branch 'main' into feature/optimize-calibrate-vocab-reading
2 parents a3bc080 + fa84955 commit cdd0be5

File tree

1 file changed

+4
-1
lines changed
  • examples/diffusers/cache_diffusion/pipeline

1 file changed

+4
-1
lines changed

examples/diffusers/cache_diffusion/pipeline/deploy.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,13 +169,16 @@ def export_onnx(backbone, onnx_path: Path):
169169
output_names = ONNX_CONFIG[backbone.__class__][f"{name}"]["output_names"]
170170
onnx_export(
171171
module,
172-
args=dummy_input,
172+
args=tuple(dummy_input.values())
173+
if isinstance(dummy_input, dict)
174+
else dummy_input,
173175
f=_onnx_file.as_posix(),
174176
input_names=input_names,
175177
output_names=output_names,
176178
dynamic_axes=ONNX_CONFIG[backbone.__class__][f"{name}"]["dynamic_axes"],
177179
do_constant_folding=True,
178180
opset_version=17,
181+
dynamo=False,
179182
)
180183
else:
181184
print(f"{_onnx_file!s} already exists!")

0 commit comments

Comments
 (0)