Upload folder using huggingface_hub

config.json

@@ -0,0 +1,9 @@
+{
+  "architectures": [
+    "HF_MODNet"
+  ],
+  "backbone": "mobilenetv2",
+  "dtype": "float32",
+  "model_type": "modnet",
+  "transformers_version": "4.57.6"
+}

configuration_modnet.py

@@ -0,0 +1,10 @@
+import torch
+from torch import nn
+from transformers import PreTrainedModel, PretrainedConfig
+
+class MODNetConfig(PretrainedConfig):
+    model_type = "modnet"
+
+    def __init__(self, backbone="mobilenetv2", **kwargs):
+        super().__init__(**kwargs)
+        self.backbone = backbone

modeling_modnet.py

@@ -0,0 +1,17 @@
+import torch
+from torch import nn
+from transformers import PreTrainedModel, PretrainedConfig
+from configuration_modnet import MODNetConfig
+
+from MODNet.modnet import MODNet
+
+
+class HF_MODNet(PreTrainedModel):
+    config_class = MODNetConfig
+
+    def __init__(self, config):
+        super().__init__(config)
+        self.modnet = MODNet(backbone_pretrained=False)
+
+    def forward(self, x, inference=True):
+        return self.modnet(x, inference)

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbbcc453dfb6aa1fd77624877c4744d27cc8cdabeb042c54513acc10a13e0fc0
+size 26284415