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