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