davidizzle commited on
Commit
ee6c71c
·
1 Parent(s): dfff032
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -25,8 +25,9 @@ gif_html.markdown(
25
  def load_model():
26
  # model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
27
  # model_id = "deepseek-ai/deepseek-llm-7b-chat"
28
- model_id = "deepseek-ai/DeepSeek-R1-Distill-Llama-70B"
29
  # model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
 
30
 
31
  # model_id = "deepseek-ai/DeepSeek-R1-Distill-Llama-8B"
32
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
25
  def load_model():
26
  # model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
27
  # model_id = "deepseek-ai/deepseek-llm-7b-chat"
28
+ # model_id = "deepseek-ai/DeepSeek-R1-Distill-Llama-70B"
29
  # model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
30
+ model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
31
 
32
  # model_id = "deepseek-ai/DeepSeek-R1-Distill-Llama-8B"
33
  tokenizer = AutoTokenizer.from_pretrained(model_id)