rphrp1985 commited on
Commit
08206a7
·
verified ·
1 Parent(s): add0b80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-32B"
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