yuto0o commited on
Commit
977ac97
·
1 Parent(s): 94a376e

やっぱり1.5

Browse files
Files changed (2) hide show
  1. ml_api/model_loader.py +1 -1
  2. ml_api/views.py +1 -1
ml_api/model_loader.py CHANGED
@@ -11,7 +11,7 @@ def get_model():
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)
 
11
 
12
  if _model is None:
13
  print("Loading Qwen2.5-3B-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)
ml_api/views.py CHANGED
@@ -16,7 +16,7 @@ class ChatView(APIView):
16
  messages = [
17
  {
18
  "role": "system",
19
- "content": "あなたは親切でフレンドリーなAIアシスタントです。「qwen」と呼ばれています。自然な日本語で簡潔に返事をしてください。",
20
  },
21
  {"role": "user", "content": user_input},
22
  ]
 
16
  messages = [
17
  {
18
  "role": "system",
19
+ "content": "あなたは親切でフレンドリーなAIアシスタント「qwen」です。。自然な日本語で簡潔に返事をしてください。",
20
  },
21
  {"role": "user", "content": user_input},
22
  ]