davidizzle commited on
Commit
ea616ac
·
1 Parent(s): 8020e39

Model upgrade

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,7 +41,8 @@ def load_model():
41
  # model_id = "google/gemma-2b-it"
42
  # tokenizer = AutoTokenizer.from_pretrained(model_id, token=True)
43
  # model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
44
- model_id = "deepseek-ai/deepseek-llm-7b-chat"
 
45
  tokenizer = AutoTokenizer.from_pretrained(model_id)
46
  model = AutoModelForCausalLM.from_pretrained(
47
  model_id,
 
41
  # model_id = "google/gemma-2b-it"
42
  # tokenizer = AutoTokenizer.from_pretrained(model_id, token=True)
43
  # model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
44
+ # model_id = "deepseek-ai/deepseek-llm-7b-chat"
45
+ model_id = "deepseek-ai/DeepSeek-V3"
46
  tokenizer = AutoTokenizer.from_pretrained(model_id)
47
  model = AutoModelForCausalLM.from_pretrained(
48
  model_id,