update interactive/config.pbtxt

InternLM · Dec 8, 2023 · b0e3984 · b0e3984
1 parent a6c4977
commit b0e3984
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lmdeploy/serve/turbomind/triton_models/interactive/config.pbtxt b/lmdeploy/serve/turbomind/triton_models/interactive/config.pbtxt
@@ -61,7 +61,7 @@ input [
   },
   {
     name: "embeddings"
-    data_type: TYPE_BYTES
+    data_type: TYPE_INT8
     dims: [ -1 ]
     optional: true
   },
@@ -72,7 +72,7 @@ input [
     reshape: { shape: [ ] }
     optional: true
   },
-    {
+  {
     name: "embedding_begins"
     data_type: TYPE_UINT32
     dims: [ -1 ]