umisetokikaze
commited on
Commit
•
6918747
1
Parent(s):
ee237a1
Update README.md
Browse files
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
|
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)
|