{
    "metadata": {
        "total_size": 767201532,
        "total_parameters": 815702965
    },
    "weight_map": {
        "s3gen.flow.decoder.estimator.down_blocks_0.downsample.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.downsample.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.down_blocks_0.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.final_block.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.final_block.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.final_block.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.final_block.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.final_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.final_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_0.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_1.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_10.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_11.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_2.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_3.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_4.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_5.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_6.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_7.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_8.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.mid_blocks_9.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.time_mlp.linear_1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.time_mlp.linear_1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.time_mlp.linear_1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.time_mlp.linear_1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.time_mlp.linear_2.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.time_mlp.linear_2.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.time_mlp.linear_2.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.time_mlp.linear_2.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.block1.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.block1.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.block1.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.block1.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.block2.conv.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.block2.conv.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.block2.norm.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.block2.norm.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.mlp_linear.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.mlp_linear.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.mlp_linear.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.mlp_linear.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.res_conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.resnet.res_conv.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_0.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_1.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_2.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.key_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.key_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.key_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.out_proj.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.out_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.out_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.out_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.query_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.query_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.query_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.value_proj.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.value_proj.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.attn.value_proj.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.ff.layers.0.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.ff.layers.0.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.ff.layers.0.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.ff.layers.0.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.ff.layers.1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.ff.layers.1.biases": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.ff.layers.1.scales": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.ff.layers.1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.norm1.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.norm1.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.norm3.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.transformer_3.norm3.weight": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.upsample.conv.bias": "model.safetensors",
        "s3gen.flow.decoder.estimator.up_blocks_0.upsample.conv.weight": "model.safetensors",
        "s3gen.flow.decoder.rand_noise": "model.safetensors",
        "s3gen.flow.encoder.after_norm.bias": "model.safetensors",
        "s3gen.flow.encoder.after_norm.weight": "model.safetensors",
        "s3gen.flow.encoder.embed.linear.bias": "model.safetensors",
        "s3gen.flow.encoder.embed.linear.biases": "model.safetensors",
        "s3gen.flow.encoder.embed.linear.scales": "model.safetensors",
        "s3gen.flow.encoder.embed.linear.weight": "model.safetensors",
        "s3gen.flow.encoder.embed.norm.bias": "model.safetensors",
        "s3gen.flow.encoder.embed.norm.weight": "model.safetensors",
        "s3gen.flow.encoder.embed.pos_enc.pe": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.encoders_0.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.encoders_1.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.encoders_2.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.encoders_3.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.encoders_4.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.encoders_5.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.pre_lookahead_layer.conv1.bias": "model.safetensors",
        "s3gen.flow.encoder.pre_lookahead_layer.conv1.weight": "model.safetensors",
        "s3gen.flow.encoder.pre_lookahead_layer.conv2.bias": "model.safetensors",
        "s3gen.flow.encoder.pre_lookahead_layer.conv2.weight": "model.safetensors",
        "s3gen.flow.encoder.up_embed.linear.bias": "model.safetensors",
        "s3gen.flow.encoder.up_embed.linear.biases": "model.safetensors",
        "s3gen.flow.encoder.up_embed.linear.scales": "model.safetensors",
        "s3gen.flow.encoder.up_embed.linear.weight": "model.safetensors",
        "s3gen.flow.encoder.up_embed.norm.bias": "model.safetensors",
        "s3gen.flow.encoder.up_embed.norm.weight": "model.safetensors",
        "s3gen.flow.encoder.up_embed.pos_enc.pe": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_0.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_1.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_2.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.feed_forward.w_1.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.feed_forward.w_1.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.feed_forward.w_1.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.feed_forward.w_1.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.feed_forward.w_2.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.feed_forward.w_2.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.feed_forward.w_2.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.feed_forward.w_2.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.norm_ff.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.norm_ff.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.norm_mha.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.norm_mha.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_k.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_k.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_k.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_k.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_out.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_out.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_out.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_out.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_pos.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_pos.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_pos.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_q.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_q.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_q.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_q.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_v.bias": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_v.biases": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_v.scales": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.linear_v.weight": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.pos_bias_u": "model.safetensors",
        "s3gen.flow.encoder.up_encoders_3.self_attn.pos_bias_v": "model.safetensors",
        "s3gen.flow.encoder.up_layer.conv.bias": "model.safetensors",
        "s3gen.flow.encoder.up_layer.conv.weight": "model.safetensors",
        "s3gen.flow.encoder_proj.bias": "model.safetensors",
        "s3gen.flow.encoder_proj.biases": "model.safetensors",
        "s3gen.flow.encoder_proj.scales": "model.safetensors",
        "s3gen.flow.encoder_proj.weight": "model.safetensors",
        "s3gen.flow.input_embedding.biases": "model.safetensors",
        "s3gen.flow.input_embedding.scales": "model.safetensors",
        "s3gen.flow.input_embedding.weight": "model.safetensors",
        "s3gen.flow.spk_embed_affine_layer.bias": "model.safetensors",
        "s3gen.flow.spk_embed_affine_layer.biases": "model.safetensors",
        "s3gen.flow.spk_embed_affine_layer.scales": "model.safetensors",
        "s3gen.flow.spk_embed_affine_layer.weight": "model.safetensors",
        "s3gen.mel2wav.conv_post.bias": "model.safetensors",
        "s3gen.mel2wav.conv_post.weight": "model.safetensors",
        "s3gen.mel2wav.conv_pre.bias": "model.safetensors",
        "s3gen.mel2wav.conv_pre.weight": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.classifier.bias": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.classifier.biases": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.classifier.scales": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.classifier.weight": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.0.bias": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.0.weight": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.1.bias": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.1.weight": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.2.bias": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.2.weight": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.3.bias": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.3.weight": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.4.bias": "model.safetensors",
        "s3gen.mel2wav.f0_predictor.condnet.4.weight": "model.safetensors",
        "s3gen.mel2wav.m_source.l_linear.bias": "model.safetensors",
        "s3gen.mel2wav.m_source.l_linear.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.0.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.1.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.2.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.3.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.4.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.5.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.6.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.7.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.resblocks.8.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.source_downs.0.bias": "model.safetensors",
        "s3gen.mel2wav.source_downs.0.weight": "model.safetensors",
        "s3gen.mel2wav.source_downs.1.bias": "model.safetensors",
        "s3gen.mel2wav.source_downs.1.weight": "model.safetensors",
        "s3gen.mel2wav.source_downs.2.bias": "model.safetensors",
        "s3gen.mel2wav.source_downs.2.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.0.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.1.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.activations1.0.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.activations1.1.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.activations1.2.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.activations2.0.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.activations2.1.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.activations2.2.alpha": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs1.0.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs1.0.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs1.1.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs1.1.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs1.2.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs1.2.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs2.0.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs2.0.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs2.1.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs2.1.weight": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs2.2.bias": "model.safetensors",
        "s3gen.mel2wav.source_resblocks.2.convs2.2.weight": "model.safetensors",
        "s3gen.mel2wav.stft_window": "model.safetensors",
        "s3gen.mel2wav.ups.0.bias": "model.safetensors",
        "s3gen.mel2wav.ups.0.weight": "model.safetensors",
        "s3gen.mel2wav.ups.1.bias": "model.safetensors",
        "s3gen.mel2wav.ups.1.weight": "model.safetensors",
        "s3gen.mel2wav.ups.2.bias": "model.safetensors",
        "s3gen.mel2wav.ups.2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.0.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.1.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.10.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.11.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.2.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.3.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.4.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.5.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.6.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.7.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.8.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.0.layers.9.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.0.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.1.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.10.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.11.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.12.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.13.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.14.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.15.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.16.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.17.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.18.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.19.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.2.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.20.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.21.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.22.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.23.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.3.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.4.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.5.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.6.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.7.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.8.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.1.layers.9.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.0.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.1.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.10.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.11.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.12.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.13.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.14.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.15.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.2.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.3.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.4.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.5.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.6.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.7.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.8.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.cam_layer.linear1.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.cam_layer.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.cam_layer.linear2.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.cam_layer.linear2.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.cam_layer.linear_local.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.linear1.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.nonlinear1.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.nonlinear1.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.nonlinear1.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.nonlinear1.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.nonlinear2.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.nonlinear2.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.nonlinear2.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.blocks.2.layers.9.nonlinear2.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.dense.linear.weight": "model.safetensors",
        "s3gen.speaker_encoder.dense.nonlinear.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.dense.nonlinear.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.bn1.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.bn1.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.bn1.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.bn1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.bn2.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.bn2.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.bn2.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.bn2.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.conv1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.conv2.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.bn1.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.bn1.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.bn1.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.bn1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.bn2.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.bn2.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.bn2.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.bn2.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.conv1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.conv2.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.shortcut.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.shortcut.1.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.shortcut.1.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.shortcut.1.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.0.shortcut.1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.bn1.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.bn1.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.bn1.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.bn1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.bn2.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.bn2.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.bn2.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.bn2.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.conv1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer1.1.conv2.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.bn1.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.bn1.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.bn1.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.bn1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.bn2.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.bn2.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.bn2.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.bn2.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.conv1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.conv2.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.shortcut.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.shortcut.1.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.shortcut.1.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.shortcut.1.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.0.shortcut.1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.bn1.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.bn1.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.bn1.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.bn1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.bn2.bias": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.bn2.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.bn2.running_var": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.bn2.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.conv1.weight": "model.safetensors",
        "s3gen.speaker_encoder.head.layer2.1.conv2.weight": "model.safetensors",
        "s3gen.speaker_encoder.out_nonlinear.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.out_nonlinear.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.out_nonlinear.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.out_nonlinear.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.tdnn.linear.weight": "model.safetensors",
        "s3gen.speaker_encoder.tdnn.nonlinear.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.tdnn.nonlinear.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.tdnn.nonlinear.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.tdnn.nonlinear.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.transits.0.linear.weight": "model.safetensors",
        "s3gen.speaker_encoder.transits.0.nonlinear.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.transits.0.nonlinear.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.transits.0.nonlinear.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.transits.0.nonlinear.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.transits.1.linear.weight": "model.safetensors",
        "s3gen.speaker_encoder.transits.1.nonlinear.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.transits.1.nonlinear.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.transits.1.nonlinear.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.transits.1.nonlinear.0.weight": "model.safetensors",
        "s3gen.speaker_encoder.transits.2.linear.weight": "model.safetensors",
        "s3gen.speaker_encoder.transits.2.nonlinear.0.bias": "model.safetensors",
        "s3gen.speaker_encoder.transits.2.nonlinear.0.running_mean": "model.safetensors",
        "s3gen.speaker_encoder.transits.2.nonlinear.0.running_var": "model.safetensors",
        "s3gen.speaker_encoder.transits.2.nonlinear.0.weight": "model.safetensors",
        "s3gen.trim_fade": "model.safetensors",
        "t3.cond_enc.emotion_adv_fc.weight": "model.safetensors",
        "t3.cond_enc.perceiver.attn.norm.bias": "model.safetensors",
        "t3.cond_enc.perceiver.attn.norm.weight": "model.safetensors",
        "t3.cond_enc.perceiver.attn.proj_out.bias": "model.safetensors",
        "t3.cond_enc.perceiver.attn.proj_out.biases": "model.safetensors",
        "t3.cond_enc.perceiver.attn.proj_out.scales": "model.safetensors",
        "t3.cond_enc.perceiver.attn.proj_out.weight": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_k.bias": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_k.biases": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_k.scales": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_k.weight": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_q.bias": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_q.biases": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_q.scales": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_q.weight": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_v.bias": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_v.biases": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_v.scales": "model.safetensors",
        "t3.cond_enc.perceiver.attn.to_v.weight": "model.safetensors",
        "t3.cond_enc.perceiver.pre_attention_query": "model.safetensors",
        "t3.cond_enc.spkr_enc.bias": "model.safetensors",
        "t3.cond_enc.spkr_enc.biases": "model.safetensors",
        "t3.cond_enc.spkr_enc.scales": "model.safetensors",
        "t3.cond_enc.spkr_enc.weight": "model.safetensors",
        "t3.speech_emb.biases": "model.safetensors",
        "t3.speech_emb.scales": "model.safetensors",
        "t3.speech_emb.weight": "model.safetensors",
        "t3.speech_head.biases": "model.safetensors",
        "t3.speech_head.scales": "model.safetensors",
        "t3.speech_head.weight": "model.safetensors",
        "t3.speech_pos_emb.emb.biases": "model.safetensors",
        "t3.speech_pos_emb.emb.scales": "model.safetensors",
        "t3.speech_pos_emb.emb.weight": "model.safetensors",
        "t3.text_emb.biases": "model.safetensors",
        "t3.text_emb.scales": "model.safetensors",
        "t3.text_emb.weight": "model.safetensors",
        "t3.text_head.biases": "model.safetensors",
        "t3.text_head.scales": "model.safetensors",
        "t3.text_head.weight": "model.safetensors",
        "t3.text_pos_emb.emb.biases": "model.safetensors",
        "t3.text_pos_emb.emb.scales": "model.safetensors",
        "t3.text_pos_emb.emb.weight": "model.safetensors",
        "t3.tfmr.lm_head.biases": "model.safetensors",
        "t3.tfmr.lm_head.scales": "model.safetensors",
        "t3.tfmr.lm_head.weight": "model.safetensors",
        "t3.tfmr.model.embed_tokens.biases": "model.safetensors",
        "t3.tfmr.model.embed_tokens.scales": "model.safetensors",
        "t3.tfmr.model.embed_tokens.weight": "model.safetensors",
        "t3.tfmr.model.layers.0.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.0.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.0.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.0.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.0.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.0.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.0.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.0.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.0.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.0.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.0.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.0.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.1.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.1.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.1.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.1.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.1.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.1.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.1.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.1.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.1.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.1.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.1.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.1.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.10.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.10.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.10.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.10.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.10.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.10.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.10.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.10.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.10.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.10.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.10.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.10.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.11.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.11.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.11.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.11.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.11.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.11.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.11.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.11.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.11.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.11.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.11.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.11.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.12.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.12.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.12.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.12.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.12.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.12.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.12.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.12.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.12.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.12.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.12.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.12.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.13.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.13.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.13.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.13.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.13.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.13.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.13.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.13.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.13.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.13.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.13.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.13.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.14.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.14.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.14.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.14.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.14.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.14.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.14.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.14.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.14.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.14.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.14.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.14.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.15.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.15.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.15.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.15.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.15.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.15.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.15.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.15.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.15.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.15.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.15.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.15.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.16.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.16.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.16.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.16.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.16.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.16.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.16.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.16.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.16.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.16.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.16.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.16.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.17.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.17.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.17.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.17.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.17.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.17.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.17.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.17.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.17.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.17.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.17.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.17.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.18.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.18.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.18.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.18.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.18.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.18.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.18.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.18.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.18.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.18.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.18.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.18.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.19.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.19.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.19.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.19.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.19.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.19.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.19.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.19.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.19.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.19.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.19.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.19.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.2.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.2.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.2.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.2.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.2.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.2.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.2.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.2.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.2.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.2.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.2.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.2.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.20.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.20.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.20.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.20.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.20.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.20.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.20.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.20.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.20.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.20.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.20.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.20.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.21.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.21.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.21.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.21.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.21.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.21.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.21.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.21.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.21.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.21.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.21.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.21.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.22.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.22.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.22.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.22.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.22.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.22.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.22.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.22.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.22.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.22.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.22.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.22.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.23.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.23.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.23.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.23.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.23.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.23.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.23.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.23.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.23.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.23.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.23.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.23.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.24.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.24.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.24.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.24.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.24.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.24.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.24.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.24.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.24.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.24.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.24.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.24.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.25.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.25.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.25.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.25.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.25.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.25.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.25.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.25.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.25.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.25.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.25.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.25.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.26.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.26.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.26.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.26.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.26.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.26.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.26.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.26.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.26.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.26.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.26.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.26.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.27.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.27.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.27.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.27.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.27.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.27.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.27.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.27.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.27.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.27.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.27.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.27.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.28.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.28.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.28.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.28.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.28.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.28.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.28.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.28.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.28.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.28.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.28.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.28.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.29.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.29.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.29.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.29.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.29.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.29.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.29.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.29.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.29.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.29.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.29.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.29.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.3.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.3.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.3.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.3.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.3.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.3.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.3.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.3.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.3.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.3.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.3.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.3.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.4.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.4.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.4.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.4.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.4.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.4.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.4.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.4.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.4.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.4.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.4.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.4.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.5.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.5.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.5.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.5.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.5.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.5.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.5.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.5.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.5.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.5.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.5.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.5.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.6.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.6.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.6.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.6.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.6.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.6.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.6.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.6.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.6.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.6.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.6.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.6.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.7.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.7.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.7.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.7.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.7.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.7.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.7.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.7.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.7.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.7.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.7.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.7.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.8.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.8.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.8.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.8.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.8.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.8.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.8.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.8.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.8.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.8.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.8.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.8.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.9.input_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.9.mlp.down_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.9.mlp.down_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.9.mlp.down_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.9.mlp.gate_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.9.mlp.gate_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.9.mlp.gate_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.9.mlp.up_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.9.mlp.up_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.9.mlp.up_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.9.post_attention_layernorm.weight": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.k_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.k_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.k_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.o_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.o_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.o_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.q_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.q_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.q_proj.weight": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.v_proj.biases": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.v_proj.scales": "model.safetensors",
        "t3.tfmr.model.layers.9.self_attn.v_proj.weight": "model.safetensors",
        "t3.tfmr.model.norm.weight": "model.safetensors",
        "ve.lstm.layers.0.Wh": "model.safetensors",
        "ve.lstm.layers.0.Wx": "model.safetensors",
        "ve.lstm.layers.0.bias": "model.safetensors",
        "ve.lstm.layers.1.Wh": "model.safetensors",
        "ve.lstm.layers.1.Wx": "model.safetensors",
        "ve.lstm.layers.1.bias": "model.safetensors",
        "ve.lstm.layers.2.Wh": "model.safetensors",
        "ve.lstm.layers.2.Wx": "model.safetensors",
        "ve.lstm.layers.2.bias": "model.safetensors",
        "ve.proj.bias": "model.safetensors",
        "ve.proj.biases": "model.safetensors",
        "ve.proj.scales": "model.safetensors",
        "ve.proj.weight": "model.safetensors",
        "ve.similarity_bias": "model.safetensors",
        "ve.similarity_weight": "model.safetensors"
    }
}