Upload folder using huggingface_hub

config.json

@@ -1,13 +1,12 @@
 {
-  "act_dim": 3,
+  "act_dim": 6,
   "action_tanh": true,
   "activation_function": "relu",
   "architectures": [
-    "HierarchicalTrainableDT"
+    "TrainableDT"
   ],
   "attn_pdrop": 0.1,
   "bos_token_id": 50256,
-  "cluster_hidden_dim": 128,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "hidden_size": 128,
@@ -15,7 +14,6 @@
   "layer_norm_epsilon": 1e-05,
   "max_ep_len": 4096,
   "model_type": "decision_transformer",
-  "n_clusters": 8,
   "n_head": 1,
   "n_inner": null,
   "n_layer": 3,
@@ -24,10 +22,9 @@
   "resid_pdrop": 0.1,
   "scale_attn_by_inverse_layer_idx": false,
   "scale_attn_weights": true,
-  "state_dim": 11,
+  "state_dim": 17,
   "torch_dtype": "float32",
   "transformers_version": "4.46.2",
   "use_cache": true,
-  "use_subgoal_weighting": true,
   "vocab_size": 1
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:19cfbea8ea33df6b0617a21144ffca3eaa1feb950d53c58f0ea1651cb432e17c
-size 5036908
+oid sha256:4f9fefae1efcb89df5ef5bdc1b71b5ace0041243c894f08e3dbb7fa8995997a2
+size 5034448