davidizzle commited on
Commit
697b3ed
·
1 Parent(s): be1328a
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ import torch
5
  # deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct
6
  # model_id = "deepseek-ai/deepseek-coder-1.3b-instruct"
7
  # model_id = "deepseek-ai/deepseek-coder-6.7b-instruct"
8
- model_id = "deepseek-ai/deepseek-coder-33b-instruct"
 
9
  # model_id = "deepseek-ai/DeepSeek-Coder-V2-Instruct"
10
  tokenizer = AutoTokenizer.from_pretrained(model_id) # Or your own!
11
  model = AutoModelForCausalLM.from_pretrained(model_id,
 
5
  # deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct
6
  # model_id = "deepseek-ai/deepseek-coder-1.3b-instruct"
7
  # model_id = "deepseek-ai/deepseek-coder-6.7b-instruct"
8
+ # model_id = "deepseek-ai/deepseek-coder-33b-instruct"
9
+ model_id = "deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct"
10
  # model_id = "deepseek-ai/DeepSeek-Coder-V2-Instruct"
11
  tokenizer = AutoTokenizer.from_pretrained(model_id) # Or your own!
12
  model = AutoModelForCausalLM.from_pretrained(model_id,