keisawada commited on
Commit
ca18bc0
·
verified ·
1 Parent(s): df24657

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -83,14 +83,14 @@ messages = [
83
  {"role": "user", "content": "ゲーム・小説・アニメに登場するアイテムボックスの特徴と、その原理を詳細に推測してください。"},
84
  ]
85
  prompt = tokenizer.apply_chat_template(
86
- messages,
87
- tokenize=False,
88
- add_generation_prompt=True
89
  )
90
  input_ids = tokenizer.encode(
91
- prompt,
92
- add_special_tokens=False,
93
- return_tensors="pt"
94
  ).to(model.device)
95
 
96
  outputs = model.generate(
 
83
  {"role": "user", "content": "ゲーム・小説・アニメに登場するアイテムボックスの特徴と、その原理を詳細に推測してください。"},
84
  ]
85
  prompt = tokenizer.apply_chat_template(
86
+ messages,
87
+ tokenize=False,
88
+ add_generation_prompt=True
89
  )
90
  input_ids = tokenizer.encode(
91
+ prompt,
92
+ add_special_tokens=False,
93
+ return_tensors="pt"
94
  ).to(model.device)
95
 
96
  outputs = model.generate(