Kendamarron
/

jimba-instruction-generator_RakutenAI-7B-instruct_lora

Model card Files Files and versions Community

Kendamarron commited on Apr 1

Commit

b9f6b59

•

1 Parent(s): 0583491

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ model.eval()
 input_text = "サンプルテキスト"
 system_message = """A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. USER: 次の要素を使った算数の文章問題を作成してください。
 要素: {user_input} ASSISTANT: """
-inputs = tokenizer(system_message.format(user_input=theme), return_tensors="pt").to(device=model.device)
 with torch.no_grad():
     outputs = model.generate(
         **inputs,

 input_text = "サンプルテキスト"
 system_message = """A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. USER: 次の要素を使った算数の文章問題を作成してください。
 要素: {user_input} ASSISTANT: """
+inputs = tokenizer(system_message.format(user_input=input_text), return_tensors="pt").to(device=model.device)
 with torch.no_grad():
     outputs = model.generate(
         **inputs,