Commit
•
6ccd171
1
Parent(s):
cf8d92b
Update README.md
Browse files
README.md
CHANGED
|
@@ -75,7 +75,7 @@ system_prompt = "あなたはプロの小説家です。\n小説を書いてく
|
|
| 75 |
|
| 76 |
prompt = input("Enter a prompt: ")
|
| 77 |
system_prompt += prompt + "\n-------- "
|
| 78 |
-
model_inputs = tokenizer([
|
| 79 |
|
| 80 |
|
| 81 |
generated_ids = model.generate(**model_inputs, max_new_tokens=new_tokens, do_sample=True)
|
|
|
|
| 75 |
|
| 76 |
prompt = input("Enter a prompt: ")
|
| 77 |
system_prompt += prompt + "\n-------- "
|
| 78 |
+
model_inputs = tokenizer([system_prompt], return_tensors="pt").to("cuda")
|
| 79 |
|
| 80 |
|
| 81 |
generated_ids = model.generate(**model_inputs, max_new_tokens=new_tokens, do_sample=True)
|