search

NVIDIA / Stable-Diffusion-WebUI-TensorRT

TensorRT Extension for Stable Diffusion Web UI
MIT License
1.92k stars 146 forks source link

[E] 3: getPluginCreator could not find plugin: InstanceNormalization_TRT version: 1 when building engine with latest TRT_VERSION="10.0.0b6" #315

Open lxe opened 7 months ago

lxe commented 7 months ago

When building engine with latest TRT_VERSION="10.0.0b6" (controlnet_v2 branch), I get this:

[E] 3: getPluginCreator could not find plugin: InstanceNormalization_TRT version: 1
[E] 3: getPluginCreator could not find plugin: InstanceNormalization_TRT version: 1
[E] ModelImporter.cpp:826: While parsing node number 26 [InstanceNormalization -> "/input_blocks.1/input_blocks.1.0/in_layers/in_layers.0/InstanceNormalization_output_0"]:
[E] ModelImporter.cpp:829: --- Begin node ---
    input: "/input_blocks.1/input_blocks.1.0/in_layers/in_layers.0/Reshape_output_0"
    input: "/input_blocks.1/input_blocks.1.0/in_layers/in_layers.0/Constant_1_output_0"
    input: "/input_blocks.1/input_blocks.1.0/in_layers/in_layers.0/Constant_2_output_0"
    output: "/input_blocks.1/input_blocks.1.0/in_layers/in_layers.0/InstanceNormalization_output_0"
    name: "/input_blocks.1/input_blocks.1.0/in_layers/in_layers.0/InstanceNormalization"
    op_type: "InstanceNormalization"
    attribute {
      name: "epsilon"
      f: 1e-05
      type: FLOAT
    }
[E] ModelImporter.cpp:832: --- End node ---
[E] ModelImporter.cpp:834: ERROR: importerUtils.cpp:1073 In function instanceNormPluginHelper:
    [8] Assertion failed: plugin != nullptr: InstanceNormalization plugin was not found in the plugin registry!
[E] In node 26 with name: /input_blocks.1/input_blocks.1.0/in_layers/in_layers.0/InstanceNormalization and operator: InstanceNormalization (instanceNormPluginHelper): UNSUPPORTED_NODE: Assertion failed: plugin != nullptr: InstanceNormalization plugin was not found in the plugin registry!
[!] Could not parse ONNX correctl

Found a workaround here: https://github.com/onnx/onnx-tensorrt/issues/597#issuecomment-922050109, and placed it here.