Update config.pbtxt
This commit is contained in:
parent
dbcd688b39
commit
77ffcb2240
@ -92,14 +92,11 @@ input [
|
|||||||
# Triton should expect to respond with a single string
|
# Triton should expect to respond with a single string
|
||||||
# output of variable length named 'text_output'
|
# output of variable length named 'text_output'
|
||||||
output [
|
output [
|
||||||
|
|
||||||
{
|
{
|
||||||
name: "text_output"
|
name: "embedding_output"
|
||||||
data_type: TYPE_STRING
|
data_type: TYPE_FP32 # 또는 TYPE_FP16
|
||||||
dims: [ -1 ]
|
dims: [ -1, -1 ] # [seq_len, hidden_size]
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
]
|
]
|
||||||
|
|
||||||
parameters: [
|
parameters: [
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user