Update config.py
Browse files
config.py
CHANGED
|
@@ -1,6 +1,16 @@
|
|
| 1 |
# config.py
|
| 2 |
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
MAX_TOKENS = 200
|
| 5 |
TEMPERATURE = 0.7
|
| 6 |
DEVICE = "cuda" # or "cpu"
|
|
|
|
| 1 |
# config.py
|
| 2 |
|
| 3 |
+
# Available models (Hugging Face or local safetensors)
|
| 4 |
+
MODEL_LIST = {
|
| 5 |
+
"ChatGpt Tune Base": "your-username/ChatGpt-Tune",
|
| 6 |
+
"Model 1": "models/model1.safetensors",
|
| 7 |
+
"Model 2": "models/model2.safetensors",
|
| 8 |
+
"Model 3": "models/model3.safetensors"
|
| 9 |
+
}
|
| 10 |
+
|
| 11 |
+
# Default model
|
| 12 |
+
DEFAULT_MODEL = "ChatGpt Tune Base"
|
| 13 |
+
|
| 14 |
MAX_TOKENS = 200
|
| 15 |
TEMPERATURE = 0.7
|
| 16 |
DEVICE = "cuda" # or "cpu"
|