extra file for printing

This commit is contained in:
Lennart J. Kurzweg (Nx2)
2024-08-08 16:52:44 +02:00
parent 76ef8502d1
commit 11f37009d3
2 changed files with 105 additions and 17 deletions

View File

@@ -9,20 +9,22 @@ from pprint import pprint
def main():
models = [
"llama3.1", # 8b
# "llama3.1:70b",
# "llama3-groq-tool-use", # latest
# "llama3-groq-tool-use:70b",
"llama3.1:70b",
"llama3-groq-tool-use", # latest
"llama3-groq-tool-use:70b",
# "mixtral:8x7b",
# "mixtral:8x22b",
"mixtral:8x22b",
# "gemma2:2b",
# "phi3", # 3.8b
# "tinyllama:1.1b",
"mistral-nemo:12b",
# "command-r-plus:104b",
]
seeds = [
2,
# 222,
# 22222,
# 2222222
222,
22222,
2222222
]
tests = [
Test(
@@ -69,15 +71,5 @@ That means If the answer is not in Chinese the answer is NOT correct! Only if th
base_url="http://bolt.hs-mittweida.de:11434"
)
print()
for hash_key in results:
result = results[hash_key]
print(f"""
\033[0;36mtest_name:\033[0m {result['test_name']}
\033[0;36mmodel:\033[0m {result['model']}
\033[0;36mseed:\033[0m {result['seed']}
\033[0;36mvalidation_result:\033[0m {result['validation']}
\033[0;36manswer: »\033[0m{result['answer']}\033[0;36m«\033[0m""")
if __name__ == "__main__":
main()