Update delta-iris/crafter/config.json

delta-iris/crafter/config.json

@@ -64,6 +64,7 @@
       [
         {
           "name": "act_emb",
+          "sub_model_name": "act_emb",
           "signature": 
           {
             "inputs": ["act"],
@@ -72,6 +73,7 @@
         },
         {
           "name": "latents_emb",
+          "sub_model_name": "latents_emb",
           "signature": 
           {
             "inputs": ["latent_tokens"],
@@ -80,6 +82,7 @@
         },
         {
           "name": "transformer",
+          "sub_model_name": "transformer",
           "signature": 
           {
             "call_mode": "auto"
@@ -87,12 +90,24 @@
         },
         {
           "name": "frame_cnn",
+          "sub_model_name": "frame_cnn",
           "signature": 
           {
             "inputs": ["obs"],
             "call_mode": "auto"
           }
         }
+      ],
+      "methods":
+      [
+        {
+          "name": "blocks_left_in_kv_cache",
+          "method_name": "blocks_left_in_kv_cache"
+        },
+        {
+          "name": "reset_kv_cache",
+          "method_name": "reset_kv_cache"
+        }
       ]
     },
     {
@@ -143,12 +158,20 @@
       [
         {
           "name": "decode",
+          "sub_model_name": "decode",
           "signature": 
           {
             "inputs": ["obs", "act", "q", "should_clamp"],
             "call_mode": "positional"
           }
         }
+      ],
+      "methods":
+      [
+        {
+          "name": "embed_tokens",
+          "method_name": "embed_tokens"
+        }
       ]
     }
   ]