{ "model_type": "LetterConditionedUnet", "num_classes": 26, "class_emb_size": 8, "sample_size": 512, "in_channels": 9, "out_channels": 1, "layers_per_block": 2, "block_out_channels": [ 64, 128, 256, 512, 512 ], "down_block_types": [ "DownBlock2D", "DownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D", "AttnDownBlock2D" ], "up_block_types": [ "AttnUpBlock2D", "AttnUpBlock2D", "AttnUpBlock2D", "UpBlock2D", "UpBlock2D" ] }