config.py

import torch

# Model Configuration
MODEL_NAME = "Kwaipilot/KAT-Dev"

# Generation Configuration
MAX_NEW_TOKENS = 1024
TEMPERATURE = 0.7
DO_SAMPLE = True