File size: 388 Bytes
9c56b78 7bf02b4 9c56b78 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# 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"
|