Update app.py
Browse files
app.py
CHANGED
@@ -119,7 +119,7 @@ import transformers
|
|
119 |
# model = load_checkpoint_and_dispatch(model, model_id, device_map=device_map, no_split_module_classes=["GPTJBlock"])
|
120 |
# model.half()
|
121 |
|
122 |
-
MODEL_ID = "deepseek-ai/DeepSeek-R1-Distill-Qwen-
|
123 |
CHAT_TEMPLATE = "َAuto"
|
124 |
MODEL_NAME = MODEL_ID.split("/")[-1]
|
125 |
CONTEXT_LENGTH = 16000
|
|
|
119 |
# model = load_checkpoint_and_dispatch(model, model_id, device_map=device_map, no_split_module_classes=["GPTJBlock"])
|
120 |
# model.half()
|
121 |
|
122 |
+
MODEL_ID = "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B"
|
123 |
CHAT_TEMPLATE = "َAuto"
|
124 |
MODEL_NAME = MODEL_ID.split("/")[-1]
|
125 |
CONTEXT_LENGTH = 16000
|