groupuser/codesage-large-v2
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-11-13 04:45:40 +00:00
parent dbcd688b39
commit 77ffcb2240
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
config.pbtxt
View File

@ -92,14 +92,11 @@ input [
# Triton should expect to respond with a single string
# output of variable length named 'text_output'
output [
{
name: "text_output"
data_type: TYPE_STRING
dims: [ -1 ]
name: "embedding_output"
data_type: TYPE_FP32 # 또는 TYPE_FP16
dims: [ -1, -1 ] # [seq_len, hidden_size]
}
]
parameters: [