From 797b72e2aeb7ef528bf61248d71805e3665b3217 Mon Sep 17 00:00:00 2001
From: Kevin Pedro <kpedro88@gmail.com>
Date: Wed, 3 Jan 2024 04:09:26 -0600
Subject: [PATCH] add model checksums

---
 models/MustacheEB/config.pbtxt | 50 ++++++++++++++++++----------------
 models/MustacheEE/config.pbtxt | 50 ++++++++++++++++++----------------
 2 files changed, 52 insertions(+), 48 deletions(-)

diff --git a/models/MustacheEB/config.pbtxt b/models/MustacheEB/config.pbtxt
index d5ef2f2..12c4961 100644
--- a/models/MustacheEB/config.pbtxt
+++ b/models/MustacheEB/config.pbtxt
@@ -1,27 +1,29 @@
 name: "MustacheEB"
 platform: "pytorch_libtorch"
-max_batch_size: 0
-input [
-{
-    name: "x__0"
-    data_type: TYPE_FP32
-    dims: [-1, 4]
-},
-{
-    name: "batch__1"
-    data_type: TYPE_INT64
-    dims: [-1]
-},
-{ 
-    name: "graphx__2"
-    data_type: TYPE_FP32
-    dims: [-1]
-}
-]
-output [
-{
-    name: "dscb__0"
-    data_type: TYPE_FP32
-    dims: [-1, 6]
+input [{
+  name: "x__0"
+  data_type: TYPE_FP32
+  dims: [-1, 4]
+}, {
+  name: "batch__1"
+  data_type: TYPE_INT64
+  dims: [-1]
+}, {
+  name: "graphx__2"
+  data_type: TYPE_FP32
+  dims: [-1]
+}]
+output [{
+  name: "dscb__0"
+  data_type: TYPE_FP32
+  dims: [-1, 6]
 }]
-
+model_repository_agents {
+  agents [{
+    name: "checksum"
+    parameters {
+      key: "MD5:1/model.pt"
+      value: "27ef08261d777827c1dd932aa69aa305"
+    }
+  }]
+}
diff --git a/models/MustacheEE/config.pbtxt b/models/MustacheEE/config.pbtxt
index 61f083e..178f5df 100644
--- a/models/MustacheEE/config.pbtxt
+++ b/models/MustacheEE/config.pbtxt
@@ -1,27 +1,29 @@
 name: "MustacheEE"
 platform: "pytorch_libtorch"
-max_batch_size: 0
-input [
-{
-    name: "x__0"
-    data_type: TYPE_FP32
-    dims: [-1, 4]
-},
-{
-    name: "batch__1"
-    data_type: TYPE_INT64
-    dims: [-1]
-},
-{ 
-    name: "graphx__2"
-    data_type: TYPE_FP32
-    dims: [-1]
-}
-]
-output [
-{
-    name: "dscb__0"
-    data_type: TYPE_FP32
-    dims: [-1, 6]
+input [{
+  name: "x__0"
+  data_type: TYPE_FP32
+  dims: [-1, 4]
+}, {
+  name: "batch__1"
+  data_type: TYPE_INT64
+  dims: [-1]
+}, {
+  name: "graphx__2"
+  data_type: TYPE_FP32
+  dims: [-1]
+}]
+output [{
+  name: "dscb__0"
+  data_type: TYPE_FP32
+  dims: [-1, 6]
 }]
-
+model_repository_agents {
+  agents [{
+    name: "checksum"
+    parameters {
+      key: "MD5:1/model.pt"
+      value: "17e8258349456152010be91b08879db8"
+    }
+  }]
+}