| { | |
| "architecture": "alucard-unet", | |
| "in_channels": 8, | |
| "out_channels": 4, | |
| "base_channels": 64, | |
| "channel_mults": [ | |
| 1, | |
| 2, | |
| 4, | |
| 4 | |
| ], | |
| "num_res_blocks": 2, | |
| "attn_resolutions": [ | |
| 32, | |
| 16 | |
| ], | |
| "text_dim": 512, | |
| "image_size": 128, | |
| "text_encoder": "openai/clip-vit-base-patch32", | |
| "parameters": 31956228, | |
| "dtype": "float16", | |
| "source_checkpoint": "checkpoints/best.pt", | |
| "source_epoch": 110, | |
| "source_step": 28638, | |
| "weights": "ema" | |
| } |