{ "architectures": [ "UNetMSS3D" ], "auto_map": { "AutoConfig": "UNetConfigs.UNetMSS3DConfig", "AutoModel": "UNets.UNetMSS3D" }, "in_ch": 1, "init_features": 64, "model_type": "UNetMSS", "out_ch": 1, "torch_dtype": "float32", "transformers_version": "4.44.2" }