mirror of
https://github.com/Ladebeze66/llm_lab_perso.git
synced 2025-12-15 19:26:52 +01:00
18 lines
563 B
Python
18 lines
563 B
Python
#Tester plusieurs agents avec les mêmes paramètres
|
|
import sys
|
|
import os
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
from core.factory import LLMFactory
|
|
from utils.parameter_tester import test_agents_on_prompt
|
|
|
|
test_agents_on_prompt(
|
|
model_class=LLMFactory._registry["mistral7b"],
|
|
prompt="Explique ce qu'est le machine learning.",
|
|
agents=["formateur", "chercheur", "assistant_technique"],
|
|
param_grid={},
|
|
fixed_params={
|
|
"temperature": 1.5,
|
|
"top_p": 0.9,
|
|
"num_predict": 300
|
|
}
|
|
) |