groupuser/base-gemma-3-1b-it
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update config.pbtxt

Browse Source
This commit is contained in:
cheetahadmin 2025-10-20 00:44:29 +00:00
parent a6ffcf8e84
commit f375148009
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
config.pbtxt
View File

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