{ "architectures": [ "HFUNetModel" ], "attention_resolutions": [ 16, 32, 64 ], "channel_mult": [ 0.5, 1, 1, 2, 2, 4, 4 ], "dropout": 0.0, "image_size": 512, "in_channels": 3, "model_channels": 256, "num_classes": null, "num_head_channels": 64, "num_heads": 4, "num_heads_upsample": -1, "num_res_blocks": 2, "out_channels": 6, "resblock_updown": true, "torch_dtype": "float32", "transformers_version": "4.22.2", "use_checkpoint": false, "use_fp16": true, "use_new_attention_order": false, "use_scale_shift_norm": true }