Fixing c_trt_args

Signed-off-by: Boris Fomitchev <[email protected]>
Project-MONAI · Nov 15, 2024 · 658e4f5 · 658e4f5
1 parent 938b6a0
commit 658e4f5
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/models/maisi_ct_generative/configs/inference_trt.json b/models/maisi_ct_generative/configs/inference_trt.json
@@ -4,18 +4,17 @@
     ],
     "c_trt_args": {
         "export_args": {
-            "dynamo": "$True",
+            "dynamo": "$False",
             "report": "$True"
         },
         "output_lists": [
             [
                 -1
             ],
             [
-                0
-            ]
+	    ]
         ]
     },
-    "controlnet": "$trt_compile(@controlnet_def.to(@device), @trained_controlnet_path)",
+    "controlnet": "$trt_compile(@controlnet_def.to(@device), @trained_controlnet_path, @c_trt_args)",
     "diffusion_unet": "$trt_compile(@diffusion_unet_def.to(@device), @trained_diffusion_path)"
 }