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