{

"time_embed.0.weight": "[1024, 256]",

"time_embed.0.bias": "[1024]",

"time_embed.2.weight": "[1024, 1024]",

"time_embed.2.bias": "[1024]",

"input_blocks.0.0.weight": "[256, 3, 3, 3]",

"input_blocks.0.0.bias": "[256]",

"input_blocks.1.0.in_layers.0.weight": "[256]",

"input_blocks.1.0.in_layers.0.bias": "[256]",

"input_blocks.1.0.in_layers.2.weight": "[256, 256, 3, 3]",

"input_blocks.1.0.in_layers.2.bias": "[256]",

"input_blocks.1.0.emb_layers.1.weight": "[512, 1024]",

"input_blocks.1.0.emb_layers.1.bias": "[512]",

"input_blocks.1.0.out_layers.0.weight": "[256]",

"input_blocks.1.0.out_layers.0.bias": "[256]",

"input_blocks.1.0.out_layers.3.weight": "[256, 256, 3, 3]",

"input_blocks.1.0.out_layers.3.bias": "[256]",

"input_blocks.2.0.in_layers.0.weight": "[256]",

"input_blocks.2.0.in_layers.0.bias": "[256]",

"input_blocks.2.0.in_layers.2.weight": "[256, 256, 3, 3]",

"input_blocks.2.0.in_layers.2.bias": "[256]",

"input_blocks.2.0.emb_layers.1.weight": "[512, 1024]",

"input_blocks.2.0.emb_layers.1.bias": "[512]",

"input_blocks.2.0.out_layers.0.weight": "[256]",

"input_blocks.2.0.out_layers.0.bias": "[256]",

"input_blocks.2.0.out_layers.3.weight": "[256, 256, 3, 3]",

"input_blocks.2.0.out_layers.3.bias": "[256]",

"input_blocks.3.0.in_layers.0.weight": "[256]",

"input_blocks.3.0.in_layers.0.bias": "[256]",

"input_blocks.3.0.in_layers.2.weight": "[256, 256, 3, 3]",

"input_blocks.3.0.in_layers.2.bias": "[256]",

"input_blocks.3.0.emb_layers.1.weight": "[512, 1024]",

"input_blocks.3.0.emb_layers.1.bias": "[512]",

"input_blocks.3.0.out_layers.0.weight": "[256]",

"input_blocks.3.0.out_layers.0.bias": "[256]",

"input_blocks.3.0.out_layers.3.weight": "[256, 256, 3, 3]",

"input_blocks.3.0.out_layers.3.bias": "[256]",

"input_blocks.4.0.in_layers.0.weight": "[256]",

"input_blocks.4.0.in_layers.0.bias": "[256]",

"input_blocks.4.0.in_layers.2.weight": "[256, 256, 3, 3]",

"input_blocks.4.0.in_layers.2.bias": "[256]",

"input_blocks.4.0.emb_layers.1.weight": "[512, 1024]",

"input_blocks.4.0.emb_layers.1.bias": "[512]",

"input_blocks.4.0.out_layers.0.weight": "[256]",

"input_blocks.4.0.out_layers.0.bias": "[256]",

"input_blocks.4.0.out_layers.3.weight": "[256, 256, 3, 3]",

"input_blocks.4.0.out_layers.3.bias": "[256]",

"input_blocks.5.0.in_layers.0.weight": "[256]",

"input_blocks.5.0.in_layers.0.bias": "[256]",

"input_blocks.5.0.in_layers.2.weight": "[256, 256, 3, 3]",

"input_blocks.5.0.in_layers.2.bias": "[256]",

"input_blocks.5.0.emb_layers.1.weight": "[512, 1024]",

"input_blocks.5.0.emb_layers.1.bias": "[512]",

"input_blocks.5.0.out_layers.0.weight": "[256]",

"input_blocks.5.0.out_layers.0.bias": "[256]",

"input_blocks.5.0.out_layers.3.weight": "[256, 256, 3, 3]",

"input_blocks.5.0.out_layers.3.bias": "[256]",

"input_blocks.6.0.in_layers.0.weight": "[256]",

"input_blocks.6.0.in_layers.0.bias": "[256]",

"input_blocks.6.0.in_layers.2.weight": "[256, 256, 3, 3]",

"input_blocks.6.0.in_layers.2.bias": "[256]",

"input_blocks.6.0.emb_layers.1.weight": "[512, 1024]",

"input_blocks.6.0.emb_layers.1.bias": "[512]",

"input_blocks.6.0.out_layers.0.weight": "[256]",

"input_blocks.6.0.out_layers.0.bias": "[256]",

"input_blocks.6.0.out_layers.3.weight": "[256, 256, 3, 3]",

"input_blocks.6.0.out_layers.3.bias": "[256]",

"input_blocks.7.0.in_layers.0.weight": "[256]",

"input_blocks.7.0.in_layers.0.bias": "[256]",

"input_blocks.7.0.in_layers.2.weight": "[512, 256, 3, 3]",

"input_blocks.7.0.in_layers.2.bias": "[512]",

"input_blocks.7.0.emb_layers.1.weight": "[1024, 1024]",

"input_blocks.7.0.emb_layers.1.bias": "[1024]",

"input_blocks.7.0.out_layers.0.weight": "[512]",

"input_blocks.7.0.out_layers.0.bias": "[512]",

"input_blocks.7.0.out_layers.3.weight": "[512, 512, 3, 3]",

"input_blocks.7.0.out_layers.3.bias": "[512]",

"input_blocks.7.0.skip_connection.weight": "[512, 256, 1, 1]",

"input_blocks.7.0.skip_connection.bias": "[512]",

"input_blocks.8.0.in_layers.0.weight": "[512]",

"input_blocks.8.0.in_layers.0.bias": "[512]",

"input_blocks.8.0.in_layers.2.weight": "[512, 512, 3, 3]",

"input_blocks.8.0.in_layers.2.bias": "[512]",

"input_blocks.8.0.emb_layers.1.weight": "[1024, 1024]",

"input_blocks.8.0.emb_layers.1.bias": "[1024]",

"input_blocks.8.0.out_layers.0.weight": "[512]",

"input_blocks.8.0.out_layers.0.bias": "[512]",

"input_blocks.8.0.out_layers.3.weight": "[512, 512, 3, 3]",

"input_blocks.8.0.out_layers.3.bias": "[512]",

"input_blocks.9.0.in_layers.0.weight": "[512]",

"input_blocks.9.0.in_layers.0.bias": "[512]",

"input_blocks.9.0.in_layers.2.weight": "[512, 512, 3, 3]",

"input_blocks.9.0.in_layers.2.bias": "[512]",

"input_blocks.9.0.emb_layers.1.weight": "[1024, 1024]",

"input_blocks.9.0.emb_layers.1.bias": "[1024]",

"input_blocks.9.0.out_layers.0.weight": "[512]",

"input_blocks.9.0.out_layers.0.bias": "[512]",

"input_blocks.9.0.out_layers.3.weight": "[512, 512, 3, 3]",

"input_blocks.9.0.out_layers.3.bias": "[512]",

"input_blocks.10.0.in_layers.0.weight": "[512]",

"input_blocks.10.0.in_layers.0.bias": "[512]",

"input_blocks.10.0.in_layers.2.weight": "[512, 512, 3, 3]",

"input_blocks.10.0.in_layers.2.bias": "[512]",

"input_blocks.10.0.emb_layers.1.weight": "[1024, 1024]",

"input_blocks.10.0.emb_layers.1.bias": "[1024]",

"input_blocks.10.0.out_layers.0.weight": "[512]",

"input_blocks.10.0.out_layers.0.bias": "[512]",

"input_blocks.10.0.out_layers.3.weight": "[512, 512, 3, 3]",

"input_blocks.10.0.out_layers.3.bias": "[512]",

"input_blocks.10.1.norm.weight": "[512]",

"input_blocks.10.1.norm.bias": "[512]",

"input_blocks.10.1.qkv.weight": "[1536, 512, 1]",

"input_blocks.10.1.qkv.bias": "[1536]",

"input_blocks.10.1.proj_out.weight": "[512, 512, 1]",

"input_blocks.10.1.proj_out.bias": "[512]",

"input_blocks.11.0.in_layers.0.weight": "[512]",

"input_blocks.11.0.in_layers.0.bias": "[512]",

"input_blocks.11.0.in_layers.2.weight": "[512, 512, 3, 3]",

"input_blocks.11.0.in_layers.2.bias": "[512]",

"input_blocks.11.0.emb_layers.1.weight": "[1024, 1024]",

"input_blocks.11.0.emb_layers.1.bias": "[1024]",

"input_blocks.11.0.out_layers.0.weight": "[512]",

"input_blocks.11.0.out_layers.0.bias": "[512]",

"input_blocks.11.0.out_layers.3.weight": "[512, 512, 3, 3]",

"input_blocks.11.0.out_layers.3.bias": "[512]",

"input_blocks.11.1.norm.weight": "[512]",

"input_blocks.11.1.norm.bias": "[512]",

"input_blocks.11.1.qkv.weight": "[1536, 512, 1]",

"input_blocks.11.1.qkv.bias": "[1536]",

"input_blocks.11.1.proj_out.weight": "[512, 512, 1]",

"input_blocks.11.1.proj_out.bias": "[512]",

"input_blocks.12.0.in_layers.0.weight": "[512]",

"input_blocks.12.0.in_layers.0.bias": "[512]",

"input_blocks.12.0.in_layers.2.weight": "[512, 512, 3, 3]",

"input_blocks.12.0.in_layers.2.bias": "[512]",

"input_blocks.12.0.emb_layers.1.weight": "[1024, 1024]",

"input_blocks.12.0.emb_layers.1.bias": "[1024]",

"input_blocks.12.0.out_layers.0.weight": "[512]",

"input_blocks.12.0.out_layers.0.bias": "[512]",

"input_blocks.12.0.out_layers.3.weight": "[512, 512, 3, 3]",

"input_blocks.12.0.out_layers.3.bias": "[512]",

"input_blocks.13.0.in_layers.0.weight": "[512]",

"input_blocks.13.0.in_layers.0.bias": "[512]",

"input_blocks.13.0.in_layers.2.weight": "[1024, 512, 3, 3]",

"input_blocks.13.0.in_layers.2.bias": "[1024]",

"input_blocks.13.0.emb_layers.1.weight": "[2048, 1024]",

"input_blocks.13.0.emb_layers.1.bias": "[2048]",

"input_blocks.13.0.out_layers.0.weight": "[1024]",

"input_blocks.13.0.out_layers.0.bias": "[1024]",

"input_blocks.13.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"input_blocks.13.0.out_layers.3.bias": "[1024]",

"input_blocks.13.0.skip_connection.weight": "[1024, 512, 1, 1]",

"input_blocks.13.0.skip_connection.bias": "[1024]",

"input_blocks.13.1.norm.weight": "[1024]",

"input_blocks.13.1.norm.bias": "[1024]",

"input_blocks.13.1.qkv.weight": "[3072, 1024, 1]",

"input_blocks.13.1.qkv.bias": "[3072]",

"input_blocks.13.1.proj_out.weight": "[1024, 1024, 1]",

"input_blocks.13.1.proj_out.bias": "[1024]",

"input_blocks.14.0.in_layers.0.weight": "[1024]",

"input_blocks.14.0.in_layers.0.bias": "[1024]",

"input_blocks.14.0.in_layers.2.weight": "[1024, 1024, 3, 3]",

"input_blocks.14.0.in_layers.2.bias": "[1024]",

"input_blocks.14.0.emb_layers.1.weight": "[2048, 1024]",

"input_blocks.14.0.emb_layers.1.bias": "[2048]",

"input_blocks.14.0.out_layers.0.weight": "[1024]",

"input_blocks.14.0.out_layers.0.bias": "[1024]",

"input_blocks.14.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"input_blocks.14.0.out_layers.3.bias": "[1024]",

"input_blocks.14.1.norm.weight": "[1024]",

"input_blocks.14.1.norm.bias": "[1024]",

"input_blocks.14.1.qkv.weight": "[3072, 1024, 1]",

"input_blocks.14.1.qkv.bias": "[3072]",

"input_blocks.14.1.proj_out.weight": "[1024, 1024, 1]",

"input_blocks.14.1.proj_out.bias": "[1024]",

"input_blocks.15.0.in_layers.0.weight": "[1024]",

"input_blocks.15.0.in_layers.0.bias": "[1024]",

"input_blocks.15.0.in_layers.2.weight": "[1024, 1024, 3, 3]",

"input_blocks.15.0.in_layers.2.bias": "[1024]",

"input_blocks.15.0.emb_layers.1.weight": "[2048, 1024]",

"input_blocks.15.0.emb_layers.1.bias": "[2048]",

"input_blocks.15.0.out_layers.0.weight": "[1024]",

"input_blocks.15.0.out_layers.0.bias": "[1024]",

"input_blocks.15.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"input_blocks.15.0.out_layers.3.bias": "[1024]",

"input_blocks.16.0.in_layers.0.weight": "[1024]",

"input_blocks.16.0.in_layers.0.bias": "[1024]",

"input_blocks.16.0.in_layers.2.weight": "[1024, 1024, 3, 3]",

"input_blocks.16.0.in_layers.2.bias": "[1024]",

"input_blocks.16.0.emb_layers.1.weight": "[2048, 1024]",

"input_blocks.16.0.emb_layers.1.bias": "[2048]",

"input_blocks.16.0.out_layers.0.weight": "[1024]",

"input_blocks.16.0.out_layers.0.bias": "[1024]",

"input_blocks.16.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"input_blocks.16.0.out_layers.3.bias": "[1024]",

"input_blocks.16.1.norm.weight": "[1024]",

"input_blocks.16.1.norm.bias": "[1024]",

"input_blocks.16.1.qkv.weight": "[3072, 1024, 1]",

"input_blocks.16.1.qkv.bias": "[3072]",

"input_blocks.16.1.proj_out.weight": "[1024, 1024, 1]",

"input_blocks.16.1.proj_out.bias": "[1024]",

"input_blocks.17.0.in_layers.0.weight": "[1024]",

"input_blocks.17.0.in_layers.0.bias": "[1024]",

"input_blocks.17.0.in_layers.2.weight": "[1024, 1024, 3, 3]",

"input_blocks.17.0.in_layers.2.bias": "[1024]",

"input_blocks.17.0.emb_layers.1.weight": "[2048, 1024]",

"input_blocks.17.0.emb_layers.1.bias": "[2048]",

"input_blocks.17.0.out_layers.0.weight": "[1024]",

"input_blocks.17.0.out_layers.0.bias": "[1024]",

"input_blocks.17.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"input_blocks.17.0.out_layers.3.bias": "[1024]",

"input_blocks.17.1.norm.weight": "[1024]",

"input_blocks.17.1.norm.bias": "[1024]",

"input_blocks.17.1.qkv.weight": "[3072, 1024, 1]",

"input_blocks.17.1.qkv.bias": "[3072]",

"input_blocks.17.1.proj_out.weight": "[1024, 1024, 1]",

"input_blocks.17.1.proj_out.bias": "[1024]",

"middle_block.0.in_layers.0.weight": "[1024]",

"middle_block.0.in_layers.0.bias": "[1024]",

"middle_block.0.in_layers.2.weight": "[1024, 1024, 3, 3]",

"middle_block.0.in_layers.2.bias": "[1024]",

"middle_block.0.emb_layers.1.weight": "[2048, 1024]",

"middle_block.0.emb_layers.1.bias": "[2048]",

"middle_block.0.out_layers.0.weight": "[1024]",

"middle_block.0.out_layers.0.bias": "[1024]",

"middle_block.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"middle_block.0.out_layers.3.bias": "[1024]",

"middle_block.1.norm.weight": "[1024]",

"middle_block.1.norm.bias": "[1024]",

"middle_block.1.qkv.weight": "[3072, 1024, 1]",

"middle_block.1.qkv.bias": "[3072]",

"middle_block.1.proj_out.weight": "[1024, 1024, 1]",

"middle_block.1.proj_out.bias": "[1024]",

"middle_block.2.in_layers.0.weight": "[1024]",

"middle_block.2.in_layers.0.bias": "[1024]",

"middle_block.2.in_layers.2.weight": "[1024, 1024, 3, 3]",

"middle_block.2.in_layers.2.bias": "[1024]",

"middle_block.2.emb_layers.1.weight": "[2048, 1024]",

"middle_block.2.emb_layers.1.bias": "[2048]",

"middle_block.2.out_layers.0.weight": "[1024]",

"middle_block.2.out_layers.0.bias": "[1024]",

"middle_block.2.out_layers.3.weight": "[1024, 1024, 3, 3]",

"middle_block.2.out_layers.3.bias": "[1024]",

"output_blocks.0.0.in_layers.0.weight": "[2048]",

"output_blocks.0.0.in_layers.0.bias": "[2048]",

"output_blocks.0.0.in_layers.2.weight": "[1024, 2048, 3, 3]",

"output_blocks.0.0.in_layers.2.bias": "[1024]",

"output_blocks.0.0.emb_layers.1.weight": "[2048, 1024]",

"output_blocks.0.0.emb_layers.1.bias": "[2048]",

"output_blocks.0.0.out_layers.0.weight": "[1024]",

"output_blocks.0.0.out_layers.0.bias": "[1024]",

"output_blocks.0.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"output_blocks.0.0.out_layers.3.bias": "[1024]",

"output_blocks.0.0.skip_connection.weight": "[1024, 2048, 1, 1]",

"output_blocks.0.0.skip_connection.bias": "[1024]",

"output_blocks.0.1.norm.weight": "[1024]",

"output_blocks.0.1.norm.bias": "[1024]",

"output_blocks.0.1.qkv.weight": "[3072, 1024, 1]",

"output_blocks.0.1.qkv.bias": "[3072]",

"output_blocks.0.1.proj_out.weight": "[1024, 1024, 1]",

"output_blocks.0.1.proj_out.bias": "[1024]",

"output_blocks.1.0.in_layers.0.weight": "[2048]",

"output_blocks.1.0.in_layers.0.bias": "[2048]",

"output_blocks.1.0.in_layers.2.weight": "[1024, 2048, 3, 3]",

"output_blocks.1.0.in_layers.2.bias": "[1024]",

"output_blocks.1.0.emb_layers.1.weight": "[2048, 1024]",

"output_blocks.1.0.emb_layers.1.bias": "[2048]",

"output_blocks.1.0.out_layers.0.weight": "[1024]",

"output_blocks.1.0.out_layers.0.bias": "[1024]",

"output_blocks.1.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"output_blocks.1.0.out_layers.3.bias": "[1024]",

"output_blocks.1.0.skip_connection.weight": "[1024, 2048, 1, 1]",

"output_blocks.1.0.skip_connection.bias": "[1024]",

"output_blocks.1.1.norm.weight": "[1024]",

"output_blocks.1.1.norm.bias": "[1024]",

"output_blocks.1.1.qkv.weight": "[3072, 1024, 1]",

"output_blocks.1.1.qkv.bias": "[3072]",

"output_blocks.1.1.proj_out.weight": "[1024, 1024, 1]",

"output_blocks.1.1.proj_out.bias": "[1024]",

"output_blocks.2.0.in_layers.0.weight": "[2048]",

"output_blocks.2.0.in_layers.0.bias": "[2048]",

"output_blocks.2.0.in_layers.2.weight": "[1024, 2048, 3, 3]",

"output_blocks.2.0.in_layers.2.bias": "[1024]",

"output_blocks.2.0.emb_layers.1.weight": "[2048, 1024]",

"output_blocks.2.0.emb_layers.1.bias": "[2048]",

"output_blocks.2.0.out_layers.0.weight": "[1024]",

"output_blocks.2.0.out_layers.0.bias": "[1024]",

"output_blocks.2.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"output_blocks.2.0.out_layers.3.bias": "[1024]",

"output_blocks.2.0.skip_connection.weight": "[1024, 2048, 1, 1]",

"output_blocks.2.0.skip_connection.bias": "[1024]",

"output_blocks.2.1.norm.weight": "[1024]",

"output_blocks.2.1.norm.bias": "[1024]",

"output_blocks.2.1.qkv.weight": "[3072, 1024, 1]",

"output_blocks.2.1.qkv.bias": "[3072]",

"output_blocks.2.1.proj_out.weight": "[1024, 1024, 1]",

"output_blocks.2.1.proj_out.bias": "[1024]",

"output_blocks.2.2.in_layers.0.weight": "[1024]",

"output_blocks.2.2.in_layers.0.bias": "[1024]",

"output_blocks.2.2.in_layers.2.weight": "[1024, 1024, 3, 3]",

"output_blocks.2.2.in_layers.2.bias": "[1024]",

"output_blocks.2.2.emb_layers.1.weight": "[2048, 1024]",

"output_blocks.2.2.emb_layers.1.bias": "[2048]",

"output_blocks.2.2.out_layers.0.weight": "[1024]",

"output_blocks.2.2.out_layers.0.bias": "[1024]",

"output_blocks.2.2.out_layers.3.weight": "[1024, 1024, 3, 3]",

"output_blocks.2.2.out_layers.3.bias": "[1024]",

"output_blocks.3.0.in_layers.0.weight": "[2048]",

"output_blocks.3.0.in_layers.0.bias": "[2048]",

"output_blocks.3.0.in_layers.2.weight": "[1024, 2048, 3, 3]",

"output_blocks.3.0.in_layers.2.bias": "[1024]",

"output_blocks.3.0.emb_layers.1.weight": "[2048, 1024]",

"output_blocks.3.0.emb_layers.1.bias": "[2048]",

"output_blocks.3.0.out_layers.0.weight": "[1024]",

"output_blocks.3.0.out_layers.0.bias": "[1024]",

"output_blocks.3.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"output_blocks.3.0.out_layers.3.bias": "[1024]",

"output_blocks.3.0.skip_connection.weight": "[1024, 2048, 1, 1]",

"output_blocks.3.0.skip_connection.bias": "[1024]",

"output_blocks.3.1.norm.weight": "[1024]",

"output_blocks.3.1.norm.bias": "[1024]",

"output_blocks.3.1.qkv.weight": "[3072, 1024, 1]",

"output_blocks.3.1.qkv.bias": "[3072]",

"output_blocks.3.1.proj_out.weight": "[1024, 1024, 1]",

"output_blocks.3.1.proj_out.bias": "[1024]",

"output_blocks.4.0.in_layers.0.weight": "[2048]",

"output_blocks.4.0.in_layers.0.bias": "[2048]",

"output_blocks.4.0.in_layers.2.weight": "[1024, 2048, 3, 3]",

"output_blocks.4.0.in_layers.2.bias": "[1024]",

"output_blocks.4.0.emb_layers.1.weight": "[2048, 1024]",

"output_blocks.4.0.emb_layers.1.bias": "[2048]",

"output_blocks.4.0.out_layers.0.weight": "[1024]",

"output_blocks.4.0.out_layers.0.bias": "[1024]",

"output_blocks.4.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"output_blocks.4.0.out_layers.3.bias": "[1024]",

"output_blocks.4.0.skip_connection.weight": "[1024, 2048, 1, 1]",

"output_blocks.4.0.skip_connection.bias": "[1024]",

"output_blocks.4.1.norm.weight": "[1024]",

"output_blocks.4.1.norm.bias": "[1024]",

"output_blocks.4.1.qkv.weight": "[3072, 1024, 1]",

"output_blocks.4.1.qkv.bias": "[3072]",

"output_blocks.4.1.proj_out.weight": "[1024, 1024, 1]",

"output_blocks.4.1.proj_out.bias": "[1024]",

"output_blocks.5.0.in_layers.0.weight": "[1536]",

"output_blocks.5.0.in_layers.0.bias": "[1536]",

"output_blocks.5.0.in_layers.2.weight": "[1024, 1536, 3, 3]",

"output_blocks.5.0.in_layers.2.bias": "[1024]",

"output_blocks.5.0.emb_layers.1.weight": "[2048, 1024]",

"output_blocks.5.0.emb_layers.1.bias": "[2048]",

"output_blocks.5.0.out_layers.0.weight": "[1024]",

"output_blocks.5.0.out_layers.0.bias": "[1024]",

"output_blocks.5.0.out_layers.3.weight": "[1024, 1024, 3, 3]",

"output_blocks.5.0.out_layers.3.bias": "[1024]",

"output_blocks.5.0.skip_connection.weight": "[1024, 1536, 1, 1]",

"output_blocks.5.0.skip_connection.bias": "[1024]",

"output_blocks.5.1.norm.weight": "[1024]",

"output_blocks.5.1.norm.bias": "[1024]",

"output_blocks.5.1.qkv.weight": "[3072, 1024, 1]",

"output_blocks.5.1.qkv.bias": "[3072]",

"output_blocks.5.1.proj_out.weight": "[1024, 1024, 1]",

"output_blocks.5.1.proj_out.bias": "[1024]",

"output_blocks.5.2.in_layers.0.weight": "[1024]",

"output_blocks.5.2.in_layers.0.bias": "[1024]",

"output_blocks.5.2.in_layers.2.weight": "[1024, 1024, 3, 3]",

"output_blocks.5.2.in_layers.2.bias": "[1024]",

"output_blocks.5.2.emb_layers.1.weight": "[2048, 1024]",

"output_blocks.5.2.emb_layers.1.bias": "[2048]",

"output_blocks.5.2.out_layers.0.weight": "[1024]",

"output_blocks.5.2.out_layers.0.bias": "[1024]",

"output_blocks.5.2.out_layers.3.weight": "[1024, 1024, 3, 3]",

"output_blocks.5.2.out_layers.3.bias": "[1024]",

"output_blocks.6.0.in_layers.0.weight": "[1536]",

"output_blocks.6.0.in_layers.0.bias": "[1536]",

"output_blocks.6.0.in_layers.2.weight": "[512, 1536, 3, 3]",

"output_blocks.6.0.in_layers.2.bias": "[512]",

"output_blocks.6.0.emb_layers.1.weight": "[1024, 1024]",

"output_blocks.6.0.emb_layers.1.bias": "[1024]",

"output_blocks.6.0.out_layers.0.weight": "[512]",

"output_blocks.6.0.out_layers.0.bias": "[512]",

"output_blocks.6.0.out_layers.3.weight": "[512, 512, 3, 3]",

"output_blocks.6.0.out_layers.3.bias": "[512]",

"output_blocks.6.0.skip_connection.weight": "[512, 1536, 1, 1]",

"output_blocks.6.0.skip_connection.bias": "[512]",

"output_blocks.6.1.norm.weight": "[512]",

"output_blocks.6.1.norm.bias": "[512]",

"output_blocks.6.1.qkv.weight": "[1536, 512, 1]",

"output_blocks.6.1.qkv.bias": "[1536]",

"output_blocks.6.1.proj_out.weight": "[512, 512, 1]",

"output_blocks.6.1.proj_out.bias": "[512]",

"output_blocks.7.0.in_layers.0.weight": "[1024]",

"output_blocks.7.0.in_layers.0.bias": "[1024]",

"output_blocks.7.0.in_layers.2.weight": "[512, 1024, 3, 3]",

"output_blocks.7.0.in_layers.2.bias": "[512]",

"output_blocks.7.0.emb_layers.1.weight": "[1024, 1024]",

"output_blocks.7.0.emb_layers.1.bias": "[1024]",

"output_blocks.7.0.out_layers.0.weight": "[512]",

"output_blocks.7.0.out_layers.0.bias": "[512]",

"output_blocks.7.0.out_layers.3.weight": "[512, 512, 3, 3]",

"output_blocks.7.0.out_layers.3.bias": "[512]",

"output_blocks.7.0.skip_connection.weight": "[512, 1024, 1, 1]",

"output_blocks.7.0.skip_connection.bias": "[512]",

"output_blocks.7.1.norm.weight": "[512]",

"output_blocks.7.1.norm.bias": "[512]",

"output_blocks.7.1.qkv.weight": "[1536, 512, 1]",

"output_blocks.7.1.qkv.bias": "[1536]",

"output_blocks.7.1.proj_out.weight": "[512, 512, 1]",

"output_blocks.7.1.proj_out.bias": "[512]",

"output_blocks.8.0.in_layers.0.weight": "[1024]",

"output_blocks.8.0.in_layers.0.bias": "[1024]",

"output_blocks.8.0.in_layers.2.weight": "[512, 1024, 3, 3]",

"output_blocks.8.0.in_layers.2.bias": "[512]",

"output_blocks.8.0.emb_layers.1.weight": "[1024, 1024]",

"output_blocks.8.0.emb_layers.1.bias": "[1024]",

"output_blocks.8.0.out_layers.0.weight": "[512]",

"output_blocks.8.0.out_layers.0.bias": "[512]",

"output_blocks.8.0.out_layers.3.weight": "[512, 512, 3, 3]",

"output_blocks.8.0.out_layers.3.bias": "[512]",

"output_blocks.8.0.skip_connection.weight": "[512, 1024, 1, 1]",

"output_blocks.8.0.skip_connection.bias": "[512]",

"output_blocks.8.1.norm.weight": "[512]",

"output_blocks.8.1.norm.bias": "[512]",

"output_blocks.8.1.qkv.weight": "[1536, 512, 1]",

"output_blocks.8.1.qkv.bias": "[1536]",

"output_blocks.8.1.proj_out.weight": "[512, 512, 1]",

"output_blocks.8.1.proj_out.bias": "[512]",

"output_blocks.8.2.in_layers.0.weight": "[512]",

"output_blocks.8.2.in_layers.0.bias": "[512]",

"output_blocks.8.2.in_layers.2.weight": "[512, 512, 3, 3]",

"output_blocks.8.2.in_layers.2.bias": "[512]",

"output_blocks.8.2.emb_layers.1.weight": "[1024, 1024]",

"output_blocks.8.2.emb_layers.1.bias": "[1024]",

"output_blocks.8.2.out_layers.0.weight": "[512]",

"output_blocks.8.2.out_layers.0.bias": "[512]",

"output_blocks.8.2.out_layers.3.weight": "[512, 512, 3, 3]",

"output_blocks.8.2.out_layers.3.bias": "[512]",

"output_blocks.9.0.in_layers.0.weight": "[1024]",

"output_blocks.9.0.in_layers.0.bias": "[1024]",

"output_blocks.9.0.in_layers.2.weight": "[512, 1024, 3, 3]",

"output_blocks.9.0.in_layers.2.bias": "[512]",

"output_blocks.9.0.emb_layers.1.weight": "[1024, 1024]",

"output_blocks.9.0.emb_layers.1.bias": "[1024]",

"output_blocks.9.0.out_layers.0.weight": "[512]",

"output_blocks.9.0.out_layers.0.bias": "[512]",

"output_blocks.9.0.out_layers.3.weight": "[512, 512, 3, 3]",

"output_blocks.9.0.out_layers.3.bias": "[512]",

"output_blocks.9.0.skip_connection.weight": "[512, 1024, 1, 1]",

"output_blocks.9.0.skip_connection.bias": "[512]",

"output_blocks.10.0.in_layers.0.weight": "[1024]",

"output_blocks.10.0.in_layers.0.bias": "[1024]",

"output_blocks.10.0.in_layers.2.weight": "[512, 1024, 3, 3]",

"output_blocks.10.0.in_layers.2.bias": "[512]",

"output_blocks.10.0.emb_layers.1.weight": "[1024, 1024]",

"output_blocks.10.0.emb_layers.1.bias": "[1024]",

"output_blocks.10.0.out_layers.0.weight": "[512]",

"output_blocks.10.0.out_layers.0.bias": "[512]",

"output_blocks.10.0.out_layers.3.weight": "[512, 512, 3, 3]",

"output_blocks.10.0.out_layers.3.bias": "[512]",

"output_blocks.10.0.skip_connection.weight": "[512, 1024, 1, 1]",

"output_blocks.10.0.skip_connection.bias": "[512]",

"output_blocks.11.0.in_layers.0.weight": "[768]",

"output_blocks.11.0.in_layers.0.bias": "[768]",

"output_blocks.11.0.in_layers.2.weight": "[512, 768, 3, 3]",

"output_blocks.11.0.in_layers.2.bias": "[512]",

"output_blocks.11.0.emb_layers.1.weight": "[1024, 1024]",

"output_blocks.11.0.emb_layers.1.bias": "[1024]",

"output_blocks.11.0.out_layers.0.weight": "[512]",

"output_blocks.11.0.out_layers.0.bias": "[512]",

"output_blocks.11.0.out_layers.3.weight": "[512, 512, 3, 3]",

"output_blocks.11.0.out_layers.3.bias": "[512]",

"output_blocks.11.0.skip_connection.weight": "[512, 768, 1, 1]",

"output_blocks.11.0.skip_connection.bias": "[512]",

"output_blocks.11.1.in_layers.0.weight": "[512]",

"output_blocks.11.1.in_layers.0.bias": "[512]",

"output_blocks.11.1.in_layers.2.weight": "[512, 512, 3, 3]",

"output_blocks.11.1.in_layers.2.bias": "[512]",

"output_blocks.11.1.emb_layers.1.weight": "[1024, 1024]",

"output_blocks.11.1.emb_layers.1.bias": "[1024]",

"output_blocks.11.1.out_layers.0.weight": "[512]",

"output_blocks.11.1.out_layers.0.bias": "[512]",

"output_blocks.11.1.out_layers.3.weight": "[512, 512, 3, 3]",

"output_blocks.11.1.out_layers.3.bias": "[512]",

"output_blocks.12.0.in_layers.0.weight": "[768]",

"output_blocks.12.0.in_layers.0.bias": "[768]",

"output_blocks.12.0.in_layers.2.weight": "[256, 768, 3, 3]",

"output_blocks.12.0.in_layers.2.bias": "[256]",

"output_blocks.12.0.emb_layers.1.weight": "[512, 1024]",

"output_blocks.12.0.emb_layers.1.bias": "[512]",

"output_blocks.12.0.out_layers.0.weight": "[256]",

"output_blocks.12.0.out_layers.0.bias": "[256]",

"output_blocks.12.0.out_layers.3.weight": "[256, 256, 3, 3]",

"output_blocks.12.0.out_layers.3.bias": "[256]",

"output_blocks.12.0.skip_connection.weight": "[256, 768, 1, 1]",

"output_blocks.12.0.skip_connection.bias": "[256]",

"output_blocks.13.0.in_layers.0.weight": "[512]",

"output_blocks.13.0.in_layers.0.bias": "[512]",

"output_blocks.13.0.in_layers.2.weight": "[256, 512, 3, 3]",

"output_blocks.13.0.in_layers.2.bias": "[256]",

"output_blocks.13.0.emb_layers.1.weight": "[512, 1024]",

"output_blocks.13.0.emb_layers.1.bias": "[512]",

"output_blocks.13.0.out_layers.0.weight": "[256]",

"output_blocks.13.0.out_layers.0.bias": "[256]",

"output_blocks.13.0.out_layers.3.weight": "[256, 256, 3, 3]",

"output_blocks.13.0.out_layers.3.bias": "[256]",

"output_blocks.13.0.skip_connection.weight": "[256, 512, 1, 1]",

"output_blocks.13.0.skip_connection.bias": "[256]",

"output_blocks.14.0.in_layers.0.weight": "[512]",

"output_blocks.14.0.in_layers.0.bias": "[512]",

"output_blocks.14.0.in_layers.2.weight": "[256, 512, 3, 3]",

"output_blocks.14.0.in_layers.2.bias": "[256]",

"output_blocks.14.0.emb_layers.1.weight": "[512, 1024]",

"output_blocks.14.0.emb_layers.1.bias": "[512]",

"output_blocks.14.0.out_layers.0.weight": "[256]",

"output_blocks.14.0.out_layers.0.bias": "[256]",

"output_blocks.14.0.out_layers.3.weight": "[256, 256, 3, 3]",

"output_blocks.14.0.out_layers.3.bias": "[256]",

"output_blocks.14.0.skip_connection.weight": "[256, 512, 1, 1]",

"output_blocks.14.0.skip_connection.bias": "[256]",

"output_blocks.14.1.in_layers.0.weight": "[256]",

"output_blocks.14.1.in_layers.0.bias": "[256]",

"output_blocks.14.1.in_layers.2.weight": "[256, 256, 3, 3]",

"output_blocks.14.1.in_layers.2.bias": "[256]",

"output_blocks.14.1.emb_layers.1.weight": "[512, 1024]",

"output_blocks.14.1.emb_layers.1.bias": "[512]",

"output_blocks.14.1.out_layers.0.weight": "[256]",

"output_blocks.14.1.out_layers.0.bias": "[256]",

"output_blocks.14.1.out_layers.3.weight": "[256, 256, 3, 3]",

"output_blocks.14.1.out_layers.3.bias": "[256]",

"output_blocks.15.0.in_layers.0.weight": "[512]",

"output_blocks.15.0.in_layers.0.bias": "[512]",

"output_blocks.15.0.in_layers.2.weight": "[256, 512, 3, 3]",

"output_blocks.15.0.in_layers.2.bias": "[256]",

"output_blocks.15.0.emb_layers.1.weight": "[512, 1024]",

"output_blocks.15.0.emb_layers.1.bias": "[512]",

"output_blocks.15.0.out_layers.0.weight": "[256]",

"output_blocks.15.0.out_layers.0.bias": "[256]",

"output_blocks.15.0.out_layers.3.weight": "[256, 256, 3, 3]",

"output_blocks.15.0.out_layers.3.bias": "[256]",

"output_blocks.15.0.skip_connection.weight": "[256, 512, 1, 1]",

"output_blocks.15.0.skip_connection.bias": "[256]",

"output_blocks.16.0.in_layers.0.weight": "[512]",

"output_blocks.16.0.in_layers.0.bias": "[512]",

"output_blocks.16.0.in_layers.2.weight": "[256, 512, 3, 3]",

"output_blocks.16.0.in_layers.2.bias": "[256]",

"output_blocks.16.0.emb_layers.1.weight": "[512, 1024]",

"output_blocks.16.0.emb_layers.1.bias": "[512]",

"output_blocks.16.0.out_layers.0.weight": "[256]",

"output_blocks.16.0.out_layers.0.bias": "[256]",

"output_blocks.16.0.out_layers.3.weight": "[256, 256, 3, 3]",

"output_blocks.16.0.out_layers.3.bias": "[256]",

"output_blocks.16.0.skip_connection.weight": "[256, 512, 1, 1]",

"output_blocks.16.0.skip_connection.bias": "[256]",

"output_blocks.17.0.in_layers.0.weight": "[512]",

"output_blocks.17.0.in_layers.0.bias": "[512]",

"output_blocks.17.0.in_layers.2.weight": "[256, 512, 3, 3]",

"output_blocks.17.0.in_layers.2.bias": "[256]",

"output_blocks.17.0.emb_layers.1.weight": "[512, 1024]",

"output_blocks.17.0.emb_layers.1.bias": "[512]",

"output_blocks.17.0.out_layers.0.weight": "[256]",

"output_blocks.17.0.out_layers.0.bias": "[256]",

"output_blocks.17.0.out_layers.3.weight": "[256, 256, 3, 3]",

"output_blocks.17.0.out_layers.3.bias": "[256]",

"output_blocks.17.0.skip_connection.weight": "[256, 512, 1, 1]",

"output_blocks.17.0.skip_connection.bias": "[256]",

"out.0.weight": "[256]",

"out.0.bias": "[256]",

"out.2.weight": "[6, 256, 3, 3]",

"out.2.bias": "[6]"

}