Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,7 @@ from threading import Thread
|
|
| 5 |
MODEL_NAMES = {
|
| 6 |
"dqnCode v0.2 1.5B": "DQN-Labs/dqnCode-v0.2-1.5B-HF",
|
| 7 |
"dqnCode v0.3 1.2B": "DQN-Labs/dqnCode-v0.3-1.2B-MLX-4bit",
|
|
|
|
| 8 |
}
|
| 9 |
model_cache = {}
|
| 10 |
def load_model(model_key):
|
|
|
|
| 5 |
MODEL_NAMES = {
|
| 6 |
"dqnCode v0.2 1.5B": "DQN-Labs/dqnCode-v0.2-1.5B-HF",
|
| 7 |
"dqnCode v0.3 1.2B": "DQN-Labs/dqnCode-v0.3-1.2B-MLX-4bit",
|
| 8 |
+
"dqnAgent": "DQN-Labs/dqnagent_v0.1_4bit"
|
| 9 |
}
|
| 10 |
model_cache = {}
|
| 11 |
def load_model(model_key):
|