soumickmj's picture
Upload ProbUNet
d341fab verified
raw
history blame contribute delete
447 Bytes
{
"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"
}