File size: 500 Bytes
02aa18d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
{
    "+imports": [
        "$from monai.networks import trt_compile"
    ],
    "c_trt_args": {
        "export_args": {
            "dynamo": "$False",
            "report": "$True"
        },
        "output_lists": [
            [
                -1
            ],
            []
        ]
    },
    "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)"
}