| { |
| "architecture": "anonymization.controlnet.ControlNet", |
| "base_model": "ffhq-diffusers", |
| "base_model_note": "Original paper FFHQ diffusion weights converted to match the Diffusers library format.", |
| "conditioning": "segmentation_masks", |
| "hint_channels": 3, |
| "down_sample_factor": 4, |
| "trained_epochs": 15, |
| "checkpoint_format": "pytorch_state_dict", |
| "checkpoint_file": "controlnet_epoch_15.pth", |
| "included_keys": [ |
| "keys/keys_CelebA_HQ.pt", |
| "keys/sub_keys_diversity_0.pt", |
| "keys/sub_keys_diversity_1.pt", |
| "keys/sub_keys_diversity_2.pt", |
| "keys/sub_keys_diversity_3.pt", |
| "keys/sub_keys_diversity_4.pt", |
| "keys/sub_keys_diversity_5.pt", |
| "keys/sub_keys_diversity_6.pt", |
| "keys/sub_keys_diversity_7.pt", |
| "keys/sub_keys_diversity_8.pt", |
| "keys/sub_keys_diversity_9.pt" |
| ] |
| } |
|
|