File size: 879 Bytes
22ca508 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
"""
Configuraci贸n del GPT Local
"""
# Configuraci贸n de modelos
DEFAULT_MODEL = "microsoft/DialoGPT-small"
AVAILABLE_MODELS = [
"microsoft/DialoGPT-small",
"gpt2",
"distilgpt2"
]
# Par谩metros de generaci贸n por defecto
DEFAULT_GENERATION_PARAMS = {
"temperature": 0.7,
"max_length": 512,
"top_p": 0.9,
"do_sample": True
}
# Configuraci贸n de la interfaz web
WEB_CONFIG = {
"host": "127.0.0.1",
"port": 7860,
"share": False,
"debug": False
}
# Configuraci贸n de logging
LOGGING_CONFIG = {
"level": "INFO",
"format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
}
# Configuraci贸n del dispositivo
DEVICE_CONFIG = {
"auto_select": True, # Seleccionar autom谩ticamente GPU si est谩 disponible
"force_cpu": False, # Forzar uso de CPU
"torch_dtype": "auto" # "float16" para GPU, "float32" para CPU
}
|