Push model using huggingface_hub.

README.md

@@ -0,0 +1,10 @@
+---
+tags:
+- model_hub_mixin
+- pytorch_model_hub_mixin
+---
+
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Code: [More Information Needed]
+- Paper: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -0,0 +1,18 @@
+{
+  "action_decoder_type": "mlp",
+  "encoder_type": "theaiinstitute/theia-base-patch16-224-cddsv",
+  "freeze_encoder": false,
+  "goal_embed": false,
+  "goal_input_dim": 2,
+  "goal_output_dim": 64,
+  "input_size": [
+    3,
+    224,
+    224
+  ],
+  "num_actions_parameters": 2,
+  "num_flatten_channels": 6,
+  "num_prediction_steps": 10,
+  "pretrained_encoder": true,
+  "state_fusion_type": "mlp"
+}

model.safetensors

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