mirrors/mindcraft
1
0
Fork 0
mirror of https://github.com/kolbytn/mindcraft.git synced 2025-09-10 12:02:59 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Edit eval script to work with ollama model

Browse source
This commit is contained in:
hlillemark 2025-03-17 11:53:31 -07:00
parent 7370161d80
commit 5a230e707e
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
evaluation_script.py
View file

@ -446,6 +446,12 @@ def make_profiles(agent_names, models, apis, template_profile="profiles/collab_p
"model": models[index],
"url": url
}
elif apis[index] == "ollama":
profile["model"] = {
"api": "ollama",
"model": models[index],
"embedding": "ollama"
}
else:
profile["model"] = models[index]