Upload folder using huggingface_hub

config.json

@@ -65,10 +65,10 @@
     ],
     "state_dim": 7,
     "action_dim": 7,
-    "d_model": 256,
+    "d_model": 512,
     "nhead": 8,
-    "num_encoder_layers": 6,
-    "num_decoder_layers": 8,
+    "num_encoder_layers": 4,
+    "num_decoder_layers": 6,
     "dim_feedforward": 512,
     "diffusion_step_embed_dim": 128,
     "down_dims": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3b582914f535f8a8d505422d5134c8acb6937446fc8ac5188c0dada80549b2dc
-size 182052508
+oid sha256:b4d5d999eaf9079c43b21383370d41ac11617e6a00fcae5006cd7c60e5299831
+size 253902700