| # config.py | |
| import torch | |
| # Device configuration | |
| DEVICE = "cuda" if torch.cuda.is_available() else "cpu" | |
| # Path to the Yi-Coder model | |
| MODEL_PATH = "path_to_yi_coder_model" # Replace with your actual model path | |
| # Maximum length for generated text | |
| MAX_LENGTH = 1024 | |
| # Temperature for text generation | |
| TEMPERATURE = 0.7 | |