File size: 334 Bytes
9463f00 27804a5 9463f00 9990e20 9463f00 27804a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# config.py
import torch
# Device configuration
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# Path to the Yi-Coder model
MODEL_PATH = "01-ai/Yi-Coder-9B-Chat" # Replace with your actual model path
# Maximum length for generated text
MAX_LENGTH = 1024
# Temperature for text generation
TEMPERATURE = 0.7 |