diff --git a/models/gat_test/config.pbtxt b/models/gat_test/config.pbtxt
index 7c823e4..434ac3a 100644
--- a/models/gat_test/config.pbtxt
+++ b/models/gat_test/config.pbtxt
@@ -1,22 +1,25 @@
 name: "gat_test"
 platform: "pytorch_libtorch"
-max_batch_size: 0
-input [
-  {
-    name: "x__0"
-    data_type: TYPE_FP32
-    dims: [ -1, 1433 ]
-  },
-  {
-    name: "edgeindex__1"
-    data_type: TYPE_INT64
-    dims: [ 2, -1 ]
-  }
-]
-output [
-  {
-    name: "logits__0"
-    data_type: TYPE_FP32
-    dims: [ -1, 7 ]
-  }
-]
+input [{
+  name: "x__0"
+  data_type: TYPE_FP32
+  dims: [-1, 1433]
+}, {
+  name: "edgeindex__1"
+  data_type: TYPE_INT64
+  dims: [2, -1]
+}]
+output [{
+  name: "logits__0"
+  data_type: TYPE_FP32
+  dims: [-1, 7]
+}]
+model_repository_agents {
+  agents [{
+    name: "checksum"
+    parameters {
+      key: "MD5:1/model.pt"
+      value: "2a887a7f45f6083b08b260158e559f95"
+    }
+  }]
+}
diff --git a/models/ragged_io/config.pbtxt b/models/ragged_io/config.pbtxt
index 13034ff..6be353c 100755
--- a/models/ragged_io/config.pbtxt
+++ b/models/ragged_io/config.pbtxt
@@ -1,29 +1,35 @@
-
 name: "ragged_io"
 platform: "onnxruntime_onnx"
-max_batch_size: 4
-version_policy: { latest { num_versions: 1 }}
-input [
-  {
-    name: "INPUT0"
-    allow_ragged_batch: true
-    data_type: TYPE_FP32
-    dims: [ -1 ]
-    
-  }
-]
-output [
-  {
-    name: "OUTPUT0"
-    data_type: TYPE_FP32
-    dims: [ -1 ]
+version_policy {
+  latest {
+    num_versions: 1
   }
-]
-batch_output [
-  {
-    target_name: "OUTPUT0"
-    kind: BATCH_SCATTER_WITH_INPUT_SHAPE
-    source_input: "INPUT0"
-  }
-]
-dynamic_batching { max_queue_delay_microseconds: 1000000 }
+}
+max_batch_size: 4
+input [{
+  name: "INPUT0"
+  data_type: TYPE_FP32
+  dims: [-1]
+  allow_ragged_batch: true
+}]
+output [{
+  name: "OUTPUT0"
+  data_type: TYPE_FP32
+  dims: [-1]
+}]
+dynamic_batching {
+  max_queue_delay_microseconds: 1000000
+}
+batch_output [{
+  target_name: ["OUTPUT0"]
+  source_input: ["INPUT0"]
+}]
+model_repository_agents {
+  agents [{
+    name: "checksum"
+    parameters {
+      key: "MD5:1/model.onnx"
+      value: "2b69fedb92d0a1440e0068dfc31cd10c"
+    }
+  }]
+}