From fa03838de390ea3aeca35e7f3ef0556c4973b2ee Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 19 Feb 2025 15:43:54 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 .../configs/inference_trt.json                | 386 +++++++++++++++++-
 1 file changed, 369 insertions(+), 17 deletions(-)

diff --git a/models/maisi_ct_generative/configs/inference_trt.json b/models/maisi_ct_generative/configs/inference_trt.json
index f2a4a8f0..d4cff79c 100644
--- a/models/maisi_ct_generative/configs/inference_trt.json
+++ b/models/maisi_ct_generative/configs/inference_trt.json
@@ -15,32 +15,384 @@
         ],
         "input_profiles": [
             {
-                "x": [[1, 4, 64, 64, 32], [1, 4, 128, 128, 128], [1, 4, 128, 128, 128]],
-                "controlnet_cond": [[1, 8, 256, 256, 128], [1, 8, 512, 512, 512], [1, 8, 512, 512, 512]]
+                "x": [
+                    [
+                        1,
+                        4,
+                        64,
+                        64,
+                        32
+                    ],
+                    [
+                        1,
+                        4,
+                        128,
+                        128,
+                        128
+                    ],
+                    [
+                        1,
+                        4,
+                        128,
+                        128,
+                        128
+                    ]
+                ],
+                "controlnet_cond": [
+                    [
+                        1,
+                        8,
+                        256,
+                        256,
+                        128
+                    ],
+                    [
+                        1,
+                        8,
+                        512,
+                        512,
+                        512
+                    ],
+                    [
+                        1,
+                        8,
+                        512,
+                        512,
+                        512
+                    ]
+                ]
             }
         ]
     },
-    "d_trt_args":{
+    "d_trt_args": {
         "export_args": {
             "dynamo": "$False",
             "do_constant_folding": "$False"
         },
         "input_profiles": [
             {
-                "x": [[1, 4, 64, 64, 32], [1, 4, 128, 128, 128], [1, 4, 128, 128, 128]],
-                "mid_block_additional_residual": [[1, 512, 8, 8, 4], [1, 512, 16, 16, 16], [1, 512, 16, 16, 16]],
-                "down_block_additional_residuals_0": [[1, 64, 64, 64, 32], [1, 64, 128, 128, 128], [1, 64, 128, 128, 128]],
-                "down_block_additional_residuals_1": [[1, 64, 64, 64, 32], [1, 64, 128, 128, 128], [1, 64, 128, 128, 128]],
-                "down_block_additional_residuals_2": [[1, 64, 64, 64, 32], [1, 64, 128, 128, 128], [1, 64, 128, 128, 128]],
-                "down_block_additional_residuals_3": [[1, 64, 32, 32, 16], [1, 64, 64, 64, 64], [1, 64, 64, 64, 64]],
-                "down_block_additional_residuals_4": [[1, 128, 32, 32, 16], [1, 128, 64, 64, 64], [1, 128, 64, 64, 64]],
-                "down_block_additional_residuals_5": [[1, 128, 32, 32, 16], [1, 128, 64, 64, 64], [1, 128, 64, 64, 64]],
-                "down_block_additional_residuals_6": [[1, 128, 16, 16, 8], [1, 128, 32, 32, 32], [1, 128, 32, 32, 32]],
-                "down_block_additional_residuals_7": [[1, 256, 16, 16, 8], [1, 256, 32, 32, 32], [1, 256, 32, 32, 32]],
-                "down_block_additional_residuals_8": [[1, 256, 16, 16, 8], [1, 256, 32, 32, 32], [1, 256, 32, 32, 32]],
-                "down_block_additional_residuals_9": [[1, 256, 8, 8, 4], [1, 256, 16, 16, 16], [1, 256, 16, 16, 16]],
-                "down_block_additional_residuals_10": [[1, 512, 8, 8, 4], [1, 512, 16, 16, 16], [1, 512, 16, 16, 16]],
-                "down_block_additional_residuals_11": [[1, 512, 8, 8, 4], [1, 512, 16, 16, 16], [1, 512, 16, 16, 16]]
+                "x": [
+                    [
+                        1,
+                        4,
+                        64,
+                        64,
+                        32
+                    ],
+                    [
+                        1,
+                        4,
+                        128,
+                        128,
+                        128
+                    ],
+                    [
+                        1,
+                        4,
+                        128,
+                        128,
+                        128
+                    ]
+                ],
+                "mid_block_additional_residual": [
+                    [
+                        1,
+                        512,
+                        8,
+                        8,
+                        4
+                    ],
+                    [
+                        1,
+                        512,
+                        16,
+                        16,
+                        16
+                    ],
+                    [
+                        1,
+                        512,
+                        16,
+                        16,
+                        16
+                    ]
+                ],
+                "down_block_additional_residuals_0": [
+                    [
+                        1,
+                        64,
+                        64,
+                        64,
+                        32
+                    ],
+                    [
+                        1,
+                        64,
+                        128,
+                        128,
+                        128
+                    ],
+                    [
+                        1,
+                        64,
+                        128,
+                        128,
+                        128
+                    ]
+                ],
+                "down_block_additional_residuals_1": [
+                    [
+                        1,
+                        64,
+                        64,
+                        64,
+                        32
+                    ],
+                    [
+                        1,
+                        64,
+                        128,
+                        128,
+                        128
+                    ],
+                    [
+                        1,
+                        64,
+                        128,
+                        128,
+                        128
+                    ]
+                ],
+                "down_block_additional_residuals_2": [
+                    [
+                        1,
+                        64,
+                        64,
+                        64,
+                        32
+                    ],
+                    [
+                        1,
+                        64,
+                        128,
+                        128,
+                        128
+                    ],
+                    [
+                        1,
+                        64,
+                        128,
+                        128,
+                        128
+                    ]
+                ],
+                "down_block_additional_residuals_3": [
+                    [
+                        1,
+                        64,
+                        32,
+                        32,
+                        16
+                    ],
+                    [
+                        1,
+                        64,
+                        64,
+                        64,
+                        64
+                    ],
+                    [
+                        1,
+                        64,
+                        64,
+                        64,
+                        64
+                    ]
+                ],
+                "down_block_additional_residuals_4": [
+                    [
+                        1,
+                        128,
+                        32,
+                        32,
+                        16
+                    ],
+                    [
+                        1,
+                        128,
+                        64,
+                        64,
+                        64
+                    ],
+                    [
+                        1,
+                        128,
+                        64,
+                        64,
+                        64
+                    ]
+                ],
+                "down_block_additional_residuals_5": [
+                    [
+                        1,
+                        128,
+                        32,
+                        32,
+                        16
+                    ],
+                    [
+                        1,
+                        128,
+                        64,
+                        64,
+                        64
+                    ],
+                    [
+                        1,
+                        128,
+                        64,
+                        64,
+                        64
+                    ]
+                ],
+                "down_block_additional_residuals_6": [
+                    [
+                        1,
+                        128,
+                        16,
+                        16,
+                        8
+                    ],
+                    [
+                        1,
+                        128,
+                        32,
+                        32,
+                        32
+                    ],
+                    [
+                        1,
+                        128,
+                        32,
+                        32,
+                        32
+                    ]
+                ],
+                "down_block_additional_residuals_7": [
+                    [
+                        1,
+                        256,
+                        16,
+                        16,
+                        8
+                    ],
+                    [
+                        1,
+                        256,
+                        32,
+                        32,
+                        32
+                    ],
+                    [
+                        1,
+                        256,
+                        32,
+                        32,
+                        32
+                    ]
+                ],
+                "down_block_additional_residuals_8": [
+                    [
+                        1,
+                        256,
+                        16,
+                        16,
+                        8
+                    ],
+                    [
+                        1,
+                        256,
+                        32,
+                        32,
+                        32
+                    ],
+                    [
+                        1,
+                        256,
+                        32,
+                        32,
+                        32
+                    ]
+                ],
+                "down_block_additional_residuals_9": [
+                    [
+                        1,
+                        256,
+                        8,
+                        8,
+                        4
+                    ],
+                    [
+                        1,
+                        256,
+                        16,
+                        16,
+                        16
+                    ],
+                    [
+                        1,
+                        256,
+                        16,
+                        16,
+                        16
+                    ]
+                ],
+                "down_block_additional_residuals_10": [
+                    [
+                        1,
+                        512,
+                        8,
+                        8,
+                        4
+                    ],
+                    [
+                        1,
+                        512,
+                        16,
+                        16,
+                        16
+                    ],
+                    [
+                        1,
+                        512,
+                        16,
+                        16,
+                        16
+                    ]
+                ],
+                "down_block_additional_residuals_11": [
+                    [
+                        1,
+                        512,
+                        8,
+                        8,
+                        4
+                    ],
+                    [
+                        1,
+                        512,
+                        16,
+                        16,
+                        16
+                    ],
+                    [
+                        1,
+                        512,
+                        16,
+                        16,
+                        16
+                    ]
+                ]
             }
         ]
     },