Create config.json

ant/weights_tensorflow_signatures/config.json

@@ -0,0 +1,44 @@
+{
+  "name": "ant_pwm_world_model",
+  "env": "Ant-v5",
+  "model_type": "pwm",
+  "metadata": {
+    "latent_dim": 32
+  },
+
+  "models": [
+    {
+      "name": "encoder",
+      "framework": null,
+      "format": "module",
+      "source": {
+        "object_ref": "encoder_saved_model"
+      },
+      "signature": {
+        "inputs": ["obs"]
+      }
+    },
+    {
+      "name": "dynamics",
+      "framework": null,
+      "format": "module",
+      "source": {
+        "object_ref": "transition_concat_saved_model"
+      },
+      "signature": {
+        "inputs": ["z", "a"]
+      }
+    },
+    {
+      "name": "reward",
+      "framework": null,
+      "format": "jit",
+      "source": {
+        "object_ref": "reward_concat_saved_model"
+      },
+      "signature": {
+        "inputs": ["z", "a"]
+      }
+    }
+  ]
+}