| { | |
| "encoder.conv_in": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.0.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.0.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.0.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.0.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.0.downsamplers.0.conv": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.1.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.1.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.1.resnets.0.conv_shortcut": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.1.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.1.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.1.downsamplers.0.conv": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.2.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.2.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.2.resnets.0.conv_shortcut": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.2.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.2.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.2.downsamplers.0.conv": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.3.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.3.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.3.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.down_blocks.3.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.mid_block.attentions.0.to_q": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.mid_block.attentions.0.to_k": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.mid_block.attentions.0.to_v": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.mid_block.attentions.0.to_out.0": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.mid_block.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.mid_block.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.mid_block.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.mid_block.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "encoder.conv_out": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.conv_in": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.0.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.0.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.0.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.0.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.0.resnets.2.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.0.resnets.2.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.0.upsamplers.0.conv": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.1.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.1.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.1.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.1.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.1.resnets.2.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.1.resnets.2.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.1.upsamplers.0.conv": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.2.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.2.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.2.resnets.0.conv_shortcut": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.2.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.2.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.2.resnets.2.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.2.resnets.2.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.2.upsamplers.0.conv": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.3.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.3.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.3.resnets.0.conv_shortcut": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.3.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.3.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.3.resnets.2.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.up_blocks.3.resnets.2.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.mid_block.attentions.0.to_q": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.mid_block.attentions.0.to_k": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.mid_block.attentions.0.to_v": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.mid_block.attentions.0.to_out.0": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.mid_block.resnets.0.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.mid_block.resnets.0.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.mid_block.resnets.1.conv1": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.mid_block.resnets.1.conv2": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| }, | |
| "decoder.conv_out": { | |
| "weights": "qint8", | |
| "activations": "none" | |
| } | |
| } |