Update mutable_torchtrt_module_example.py (#3519)

cehongwang · web-flow · commit f65c9dfa6891 · 2025-05-13T19:25:59.000+08:00
diff --git a/examples/dynamo/mutable_torchtrt_module_example.py b/examples/dynamo/mutable_torchtrt_module_example.py
@@ -106,6 +106,7 @@
             "text_embeds": {0: BATCH},
             "time_ids": {0: BATCH},
         },
+        "return_dict": False,
     }
     pipe.unet.set_expected_dynamic_shape_range(
         args_dynamic_shapes, kwargs_dynamic_shapes

Original file line number	Diff line number	Diff line change
`@@ -106,6 +106,7 @@`
`106`	`106`	`"text_embeds": {0: BATCH},`
`107`	`107`	`"time_ids": {0: BATCH},`
`108`	`108`	`},`
	`109`	`+ "return_dict": False,`
`109`	`110`	`}`
`110`	`111`	`pipe.unet.set_expected_dynamic_shape_range(`
`111`	`112`	`args_dynamic_shapes, kwargs_dynamic_shapes`