apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix bug

Browse source
This commit is contained in:
cotran 2024-11-22 11:24:27 -08:00
parent abfc81b0e7
commit b91b776874
2 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
model_server/app/commons/constants.py
View file

@ -19,6 +19,7 @@ arch_function_generation_params = {
"top_k": 50,
"max_tokens": 512,
"stop_token_ids": [151645],
"top_logprobs": 10,
}
arch_guard_model_type = {

2
model_server/app/function_calling/hallucination_handler.py
View file

@ -161,7 +161,7 @@ class HallucinationStateHandler:
self.mask.append("v")
if (
len(self.mask) > 1
and self.mask[-2] == "v"
and self.mask[-2] != "v"
and not check_parameter_property(
self.function_properties[self.function_name],
self.parameter_name[-1],