{ "architectures": [ "UNetFTSR" ], "auto_map": { "AutoConfig": "UNetFTSRConfig.UNetFTSRConfig", "AutoModel": "UNetFTSR.UNetFTSR" }, "batch_norm": false, "depth": 3, "dropout": false, "in_channels": 1, "model_type": "UNet", "n_classes": 1, "padding": true, "torch_dtype": "float32", "transformers_version": "4.44.2", "up_mode": "upconv", "wf": 6 }