Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ import torch
|
|
| 4 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 5 |
|
| 6 |
# Model configuration
|
| 7 |
-
MODEL_ID = "deepseek-ai/DeepSeek-R1-Distill-
|
| 8 |
|
| 9 |
# Define prompts and their corresponding chain-of-thought prefixes
|
| 10 |
PROMPT_COT_MAP = {
|
|
|
|
| 4 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 5 |
|
| 6 |
# Model configuration
|
| 7 |
+
MODEL_ID = "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B"
|
| 8 |
|
| 9 |
# Define prompts and their corresponding chain-of-thought prefixes
|
| 10 |
PROMPT_COT_MAP = {
|