From de5fd9599292193008acee32d65d29755fbbf796 Mon Sep 17 00:00:00 2001
From: Adrian Wang <123616592+cehongwang@users.noreply.github.com>
Date: Tue, 13 May 2025 12:47:24 +0800
Subject: [PATCH] Update mutable_torchtrt_module_example.py

---
 examples/dynamo/mutable_torchtrt_module_example.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/dynamo/mutable_torchtrt_module_example.py b/examples/dynamo/mutable_torchtrt_module_example.py
index f264b8a8d3..a6c8a5384e 100644
--- 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