Skip to content

Commit 2664241

Browse files
author
pytorchbot
committed
2025-05-13 nightly release (f65c9df)
1 parent c8a2a0e commit 2664241

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/dynamo/mutable_torchtrt_module_example.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@
106106
"text_embeds": {0: BATCH},
107107
"time_ids": {0: BATCH},
108108
},
109+
"return_dict": False,
109110
}
110111
pipe.unet.set_expected_dynamic_shape_range(
111112
args_dynamic_shapes, kwargs_dynamic_shapes

0 commit comments

Comments
 (0)