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 |
|