diff --git a/config.pbtxt b/config.pbtxt
index 4718273..bbd5c25 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -71,7 +71,7 @@ input [
   {
     name: "repetition_penalty-modify"
     data_type: TYPE_FP32
-    dims: [ 0.8 ]
+    dims: [ 2 ]
 
     optional: true