{ "architectures": [ "WNetMSS3D" ], "auto_map": { "AutoConfig": "WNetConfigs.WNetMSS3DConfig", "AutoModel": "WNets.WNetMSS3D" }, "in_ch": 1, "init_features": 64, "model_type": "WNetMSS", "out_ch": 5, "torch_dtype": "float32", "transformers_version": "4.44.2" }