yuto0o commited on
Commit
a8fd17e
·
1 Parent(s): e1bda81

gas の問題だったので3b戻す

Browse files
Files changed (1) hide show
  1. ml_api/model_loader.py +2 -2
ml_api/model_loader.py CHANGED
@@ -10,8 +10,8 @@ def get_model():
10
  global _model, _tokenizer
11
 
12
  if _model is None:
13
- print("Loading Qwen2.5-1.5B-Instruct... (Lazy Loading)")
14
- model_name = "Qwen/Qwen2.5-1.5B-Instruct"
15
  # model_name = "Qwen/Qwen2.5-1.5B-Instruct" # はやい
16
 
17
  _tokenizer = AutoTokenizer.from_pretrained(model_name)
 
10
  global _model, _tokenizer
11
 
12
  if _model is None:
13
+ print("Loading Qwen2.5-3B-Instruct... (Lazy Loading)")
14
+ model_name = "Qwen/Qwen2.5-3B-Instruct"
15
  # model_name = "Qwen/Qwen2.5-1.5B-Instruct" # はやい
16
 
17
  _tokenizer = AutoTokenizer.from_pretrained(model_name)