UD5 · Validación de la capacidad operativa | MP04 · Infraestructura para la ejecución de LLMs
Herramientas de prueba de carga
Locust (Python)
from locust import HttpUser, task, between
import json, random
PROMPTS = [
"Explica qué es la atención multi-cabeza en un transformer.",
"Describe los pasos para cuantizar un modelo LLM a INT4.",
"Diferencia entre VRAM y RAM en el contexto de la inferencia de LLMs.",
]
class LLMUser(HttpUser):
wait_time = between(1, 3)
@task
def generate(self):
self.client.post(
"/v1/chat/completions",
headers={"Authorization": "Bearer test-key"},
json={
"model": "llama-3.1-8b-instruct",
"messages": [{"role": "user", "content": random.choice(PROMPTS)}],
"max_tokens": 200,
},
timeout=60,
)
locust -f locustfile.py --host http://localhost:8000 \
-u 20 -r 2 --run-time 5m --headless --csv=resultados_carga