| { | |
| "_name_or_path": "/LocalData1/jsyoon/pnpxai_test/tutorials/conditional_unet_model/generator/258", | |
| "architectures": [ | |
| "ConditionalUNet" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration_conditional_unet.ConditionalUNetConfig", | |
| "AutoModel": "modeling_conditional_unet.ConditionalUNet" | |
| }, | |
| "encoder_rep": "evanrsl/resnet-Alzheimer", | |
| "id2label": { | |
| "0": "LABEL_0", | |
| "1": "LABEL_1", | |
| "2": "LABEL_2", | |
| "3": "LABEL_3" | |
| }, | |
| "label2id": { | |
| "LABEL_0": 0, | |
| "LABEL_1": 1, | |
| "LABEL_2": 2, | |
| "LABEL_3": 3 | |
| }, | |
| "model_type": "conditional-unet", | |
| "num_channels": 3, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.46.2" | |
| } | |