Upload folder using huggingface_hub

config.json

@@ -53,11 +53,11 @@
     "pretrained_backbone_weights": "ResNet34_Weights.IMAGENET1K_V1",
     "state_dim": 7,
     "action_dim": 7,
-    "d_model": 256,
+    "d_model": 512,
     "nhead": 8,
-    "num_encoder_layers": 4,
+    "num_encoder_layers": 8,
     "dim_feedforward": 512,
-    "diffusion_step_embed_dim": 128,
+    "diffusion_step_embed_dim": 256,
     "down_dims": [
         512,
         1024,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e575c378063bf8392c6677c7782730fb62c8e996e0612b58884199368e8045c3
-size 1201650364
+oid sha256:6300a5f75abae821587f85a4ec032ed0845e481d7bc976f701a35cec696e0931
+size 1291702148