guoday commited on
Commit
a55f745
1 Parent(s): ba5fe92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ if not torch.cuda.is_available():
23
 
24
 
25
  if torch.cuda.is_available():
26
- model_id = "deepseek-ai/deepseek-coder-1.3b-instruct"
27
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
28
  tokenizer = AutoTokenizer.from_pretrained(model_id)
29
  tokenizer.use_default_system_prompt = False
 
23
 
24
 
25
  if torch.cuda.is_available():
26
+ model_id = "deepseek-ai/deepseek-coder-6.7b-instruct"
27
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
28
  tokenizer = AutoTokenizer.from_pretrained(model_id)
29
  tokenizer.use_default_system_prompt = False