Update delta-iris/crafter/config.json

delta-iris/crafter/config.json

@@ -7,7 +7,7 @@
 
   "models": [
     {
-      "name": "tokenizer",
+      "name": "world_model",
       "framework": null,
       "format": "state_dict",
       "source": {
@@ -16,29 +16,20 @@
         "class_name": "WorldModel"
       },
       "signature": {
-        "inputs": ["obs"]
+        "call_mode": "auto"
       }
     },
     {
-      "name": "dynamics",
-      "framework": null,
-      "format": "jit",
-      "source": {
-        "path": "transition.pt"
-      },
-      "signature": {
-        "inputs": ["z", "a"]
-      }
-    },
-    {
-      "name": "reward",
+      "name": "tokenizer",
       "framework": null,
-      "format": "jit",
+      "format": "state_dict",
       "source": {
-        "path": "reward.pt"
+        "weights_path": "last.pt",
+        "class_path": "../src/tokenizer.py",
+        "class_name": "Tokenizer"
       },
       "signature": {
-        "inputs": ["z", "a"]
+        "call_mode": "auto"
       }
     }
   ]