# config.py

import torch

# Device configuration
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