Skip to content

Commit 7bf4dca

Browse files
committed
update
1 parent 08c87be commit 7bf4dca

File tree

1 file changed

+4
-0
lines changed
  • src/triton_cli/templates/llmapi/1

1 file changed

+4
-0
lines changed

src/triton_cli/templates/llmapi/1/model.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,9 @@ def _get_sampling_config_from_request(self, request):
142142
None if kwargs["top_p"] is None or kwargs["top_p"] <= 0 else kwargs["top_p"]
143143
)
144144

145+
# Remove None values
146+
kwargs = {k: v for k, v in kwargs.items() if v is not None}
147+
145148
return kwargs
146149

147150
@classmethod
@@ -244,6 +247,7 @@ def _auto_complete_inputs_and_outputs(auto_complete_model_config):
244247
"name": "max_tokens",
245248
"data_type": "TYPE_INT32",
246249
"dims": [1],
250+
"optional": True,
247251
},
248252
{
249253
"name": "stop",

0 commit comments

Comments
 (0)