Skip to content

Commit

Permalink
fix(payload): fix wrong form data key (#503)
Browse files Browse the repository at this point in the history
Because

- form data keys does not match protobuf definitions

This commit

- fix wrong form data key
  • Loading branch information
heiruwu committed Jan 30, 2024
1 parent 0ba505b commit 4d69e5e
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions pkg/handler/payload.go
Original file line number Diff line number Diff line change
Expand Up @@ -801,9 +801,9 @@ func parseTextFormDataTextGenerationInputs(req *http.Request) (textGeneration *t
if len(prompts) != 1 {
return nil, fmt.Errorf("only support batchsize 1")
}
maxNewTokenInput := req.MultipartForm.Value["max_new_token"]
maxNewTokenInput := req.MultipartForm.Value["max_new_tokens"]
temperatureInput := req.MultipartForm.Value["temperature"]
topKInput := req.MultipartForm.Value["topk"]
topKInput := req.MultipartForm.Value["top_k"]
seedInput := req.MultipartForm.Value["seed"]
extraParamsInput := req.MultipartForm.Value["extra_params"]
chatHistoryInput := req.MultipartForm.Value["chat_history"]
Expand Down Expand Up @@ -887,9 +887,9 @@ func parseTextFormDataTextGenerationChatInputs(req *http.Request) (textGeneratio
if len(prompts) != 1 {
return nil, fmt.Errorf("only support batchsize 1")
}
maxNewTokenInput := req.MultipartForm.Value["max_new_token"]
maxNewTokenInput := req.MultipartForm.Value["max_new_tokens"]
temperatureInput := req.MultipartForm.Value["temperature"]
topKInput := req.MultipartForm.Value["topk"]
topKInput := req.MultipartForm.Value["top_k"]
seedInput := req.MultipartForm.Value["seed"]
extraParamsInput := req.MultipartForm.Value["extra_params"]
chatHistoryInput := req.MultipartForm.Value["chat_history"]
Expand Down Expand Up @@ -972,9 +972,9 @@ func parseTextFormDataVisualQuestionAnsweringInputs(req *http.Request) (visualQu
if len(prompts) != 1 {
return nil, fmt.Errorf("only support batchsize 1")
}
maxNewTokenInput := req.MultipartForm.Value["max_new_token"]
maxNewTokenInput := req.MultipartForm.Value["max_new_tokens"]
temperatureInput := req.MultipartForm.Value["temperature"]
topKInput := req.MultipartForm.Value["topk"]
topKInput := req.MultipartForm.Value["top_k"]
seedInput := req.MultipartForm.Value["seed"]
extraParamsInput := req.MultipartForm.Value["extra_params"]
chatHistoryInput := req.MultipartForm.Value["chat_history"]
Expand Down

0 comments on commit 4d69e5e

Please sign in to comment.