nano6626
/

testing_unet_upload

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions

nano6626 commited on Dec 18, 2025

Commit

e34361f

·

verified ·

1 Parent(s): 38b1ab3

Update config.json

Files changed (1) hide show

config.json +14 -13

config.json CHANGED Viewed

@@ -1,14 +1,15 @@
-{
-  "N": 2,
-  "catorig": false,
-  "n_channels": 1,
-  "n_classes": 2,
-  "outker": 1,
-  "skip": true,
-  "skipcat": false,
-  "width": 32,
-  "model_type": "custom_unet",
-  "auto_map": {
-    "AutoModel": "unet.UNet"
-  }
 }

+{
+  "N": 2,
+  "catorig": false,
+  "n_channels": 1,
+  "n_classes": 2,
+  "outker": 1,
+  "skip": true,
+  "skipcat": false,
+  "width": 32,
+  "model_type": "custom_unet",
+  "auto_map": {
+    "AutoConfig": "unet.UNetConfig",
+    "AutoModel": "unet.UNet"
+  }
 }