UD4 · Puesta en servicio del modelo | MP04 · Infraestructura para la ejecución de LLMs
Parámetros de ejecución — configuración completa
from vllm import LLM, SamplingParams
SERVER_CONFIG = {
"model": "/models/llama-3.1-8b-instruct",
"dtype": "bfloat16",
"max_model_len": 8192,
"max_num_seqs": 32,
"gpu_memory_utilization": 0.90,
"tensor_parallel_size": 1,
"enforce_eager": False,
}
GENERATION_DEFAULTS = {
"max_tokens": 1024,
"temperature": 0.7,
"top_p": 0.9,
"stop": ["</s>", "[INST]"],
"repetition_penalty": 1.1,
}