Update README.md
Browse files
README.md
CHANGED
@@ -145,7 +145,6 @@ inputs = processor(
|
|
145 |
text=[text],
|
146 |
padding=True,
|
147 |
return_tensors="pt",
|
148 |
-
add_generation_prompt=True,
|
149 |
).to("cuda")
|
150 |
|
151 |
# Decoding parameters (top_p, temperature, top_k, repetition_penalty) should be tuned depending on the generation task.
|
@@ -212,7 +211,6 @@ inputs = processor(
|
|
212 |
text=[text],
|
213 |
padding=True,
|
214 |
return_tensors="pt",
|
215 |
-
add_generation_prompt=True,
|
216 |
).to("cuda")
|
217 |
|
218 |
|
|
|
145 |
text=[text],
|
146 |
padding=True,
|
147 |
return_tensors="pt",
|
|
|
148 |
).to("cuda")
|
149 |
|
150 |
# Decoding parameters (top_p, temperature, top_k, repetition_penalty) should be tuned depending on the generation task.
|
|
|
211 |
text=[text],
|
212 |
padding=True,
|
213 |
return_tensors="pt",
|
|
|
214 |
).to("cuda")
|
215 |
|
216 |
|