keisawada commited on
Commit
41c6fb7
·
verified ·
1 Parent(s): ff95d7b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -61,14 +61,14 @@ messages = [
61
  {"role": "user", "content": "ゲーム・小説・アニメに登場するアイテムボックスの特徴と、その原理を詳細に推測してください。"},
62
  ]
63
  prompt = tokenizer.apply_chat_template(
64
- messages,
65
- tokenize=False,
66
- add_generation_prompt=True
67
  )
68
  input_ids = tokenizer.encode(
69
- prompt,
70
- add_special_tokens=False,
71
- return_tensors="pt"
72
  ).to(model.device)
73
 
74
  outputs = model.generate(
 
61
  {"role": "user", "content": "ゲーム・小説・アニメに登場するアイテムボックスの特徴と、その原理を詳細に推測してください。"},
62
  ]
63
  prompt = tokenizer.apply_chat_template(
64
+ messages,
65
+ tokenize=False,
66
+ add_generation_prompt=True
67
  )
68
  input_ids = tokenizer.encode(
69
+ prompt,
70
+ add_special_tokens=False,
71
+ return_tensors="pt"
72
  ).to(model.device)
73
 
74
  outputs = model.generate(