File size: 447 Bytes
d341fab |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
{
"architectures": [
"ProbUNet"
],
"auto_map": {
"AutoConfig": "PULASkiConfigs.ProbUNetConfig",
"AutoModel": "PULASki.ProbUNet"
},
"depth": 5,
"dim": 2,
"in_channels": 1,
"latent_distribution": "normal",
"latent_size": 3,
"model_type": "ProbUNet",
"no_outact_op": false,
"num_feature_maps": 24,
"out_channels": 1,
"prob_injection_at": "end",
"torch_dtype": "float32",
"transformers_version": "4.44.2"
}
|