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 some bugs

Browse source
This commit is contained in:
Adil Hafeez 2024-10-28 20:55:58 -07:00
parent 662a840ac5
commit eaa99259ad
4 changed files with 20 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

4
model_server/app/main.py
View file

@ -186,8 +186,8 @@ async def hallucination(req: HallucinationRequest, res: Response):
start_time = time.perf_counter()
classifier = zero_shot_model["pipeline"]
if "arch_messages" in req.parameters:
req.parameters.pop("arch_messages")
if "messages" in req.parameters:
req.parameters.pop("messages")
candidate_labels = {f"{k} is {v}": k for k, v in req.parameters.items()}