prostochel097
/

alphaDepth

Depth Estimation

feature-extraction

computer-vision

Model card Files Files and versions

prostochel097 commited on Jan 5

Commit

f28ff29

·

verified ·

1 Parent(s): 7ff512a

Update config.json

Files changed (1) hide show

config.json +11 -11

config.json CHANGED Viewed

@@ -1,13 +1,13 @@
 {
-    "architecture": "AlphaUNet",
-    "version": "1.0",
-    "input_size": [
-        3,
-        128,
-        128
-    ],
-    "output_channels": 1,
-    "author": "AlphaDev",
-    "framework": "pytorch",
-    "license": "MIT"
 }

 {
+  "architectures": [
+    "AlphaUNet"
+  ],
+  "model_type": "alpha-depth",
+  "auto_map": {
+    "AutoConfig": "model.AlphaDepthConfig",
+    "AutoModel": "model.AlphaUNet"
+  },
+  "input_size": [3, 128, 128],
+  "hidden_channels": 32,
+  "transformers_version": "4.35.0"
 }