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