File size: 361 Bytes
9463f00 8a4a425 27804a5 9463f00 8a4a425 28dea9b 9463f00 8a4a425 9463f00 8a4a425 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
# config.py
import torch
# Configuración del dispositivo
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# Ruta al modelo Yi-Coder
MODEL_PATH = "01-ai/Yi-Coder-9B-Chat" # Reemplaza con la ruta real de tu modelo
# Longitud máxima para el texto generado
MAX_LENGTH = 1024
# Temperatura para la generación de texto
TEMPERATURE = 0.7
|