Upload folder using huggingface_hub

config.json

@@ -1,7 +1,13 @@
 {
-  "n_obs_steps": 1,
   "chunk_size": 100,
-  "n_action_steps": 100,
+  "dim_feedforward": 3200,
+  "dim_model": 512,
+  "dropout": 0.1,
+  "feedforward_activation": "relu",
+  "input_normalization_modes": {
+    "observation.images.top": "mean_std",
+    "observation.state": "mean_std"
+  },
   "input_shapes": {
     "observation.images.top": [
       3,
@@ -12,32 +18,26 @@
       14
     ]
   },
+  "kl_weight": 10.0,
+  "latent_dim": 32,
+  "n_action_steps": 100,
+  "n_decoder_layers": 1,
+  "n_encoder_layers": 4,
+  "n_heads": 8,
+  "n_obs_steps": 1,
+  "n_vae_encoder_layers": 4,
+  "output_normalization_modes": {
+    "action": "mean_std"
+  },
   "output_shapes": {
     "action": [
       14
     ]
   },
-  "input_normalization_modes": {
-    "observation.images.top": "mean_std",
-    "observation.state": "mean_std"
-  },
-  "output_normalization_modes": {
-    "action": "mean_std"
-  },
-  "vision_backbone": "resnet18",
+  "pre_norm": false,
   "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
   "replace_final_stride_with_dilation": false,
-  "pre_norm": false,
-  "dim_model": 512,
-  "n_heads": 8,
-  "dim_feedforward": 3200,
-  "feedforward_activation": "relu",
-  "n_encoder_layers": 4,
-  "n_decoder_layers": 1,
-  "use_vae": true,
-  "latent_dim": 32,
-  "n_vae_encoder_layers": 4,
   "temporal_ensemble_coeff": null,
-  "dropout": 0.1,
-  "kl_weight": 10.0
+  "use_vae": true,
+  "vision_backbone": "resnet18"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:02abc71850a4a301dcc114856c14c6cb623200ab0d324f0ab167cd8a300a381b
-size 206766568
+oid sha256:8a39f5bec52e202dde5fa5639e2819d2ae5935436cfbafdd5fa6015880c01a02
+size 206766560