| # config.py | |
| # Available models (Hugging Face or local safetensors) | |
| MODEL_LIST = { | |
| "ChatGpt Tune Base": "your-username/ChatGpt-Tune", | |
| "Model 1": "models/model1.safetensors", | |
| "Model 2": "models/model2.safetensors", | |
| "Model 3": "models/model3.safetensors" | |
| } | |
| # Default model | |
| DEFAULT_MODEL = "ChatGpt Tune Base" | |
| MAX_TOKENS = 200 | |
| TEMPERATURE = 0.7 | |
| DEVICE = "cuda" # or "cpu" | |