umisetokikaze commited on
Commit
6918747
1 Parent(s): ee237a1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -72,7 +72,7 @@ system_prompt = "あなたはプロの小説家です。\n小説を書いてく
72
 
73
  prompt = input("Enter a prompt: ")
74
  system_prompt += prompt + "\n-------- "
75
- model_inputs = tokenizer([system_prompt ], return_tensors="pt").to("cuda")
76
 
77
 
78
  generated_ids = model.generate(**model_inputs, max_new_tokens=new_tokens, do_sample=True)
 
72
 
73
  prompt = input("Enter a prompt: ")
74
  system_prompt += prompt + "\n-------- "
75
+ model_inputs = tokenizer([system_prompt], return_tensors="pt").to("cuda")
76
 
77
 
78
  generated_ids = model.generate(**model_inputs, max_new_tokens=new_tokens, do_sample=True)