diff --git a/config.pbtxt b/config.pbtxt
index 0e91de5..71ef2ce 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -1,5 +1,6 @@
 # Triton backend to use
-backend: "python"
+backend: "pytorch"
+platform: "pytorch_libtorch"
 max_batch_size: 0
 
 input [
@@ -7,6 +8,11 @@ input [
     name: "regen"
     data_type: TYPE_STRING
     dims: [ -1 ]
+  },
+  {
+    name: "new-regen-regi"
+    data_type: TYPE_STRING
+    dims: [ -1 ]
   }
 ]
 
@@ -14,12 +20,6 @@ output [
 ]
 
 
-parameters: [
-  {
-    key: "model_path",
-    value: {string_value: "/cheetah/input/model/groupuser/triton-python-model"}
-  }
-]
 
 
 instance_group [