diff --git a/config.pbtxt b/config.pbtxt
index 8ba7061..6a09924 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -1,7 +1,7 @@
 # Triton backend to use
 name: "base-gemma-3-1b-it"
 backend: "python"
-max_batch_size: 2
+max_batch_size: 4
 
 # Triton should expect as input a single string
 # input of variable length named 'text_input'
@@ -125,8 +125,8 @@ parameters: [
 
 instance_group [
   {
-    kind: KIND_GPU
-    count: 2
+    kind: KIND_CPU
+    count: 4
   }
 ]