Update generate.py
Browse files- generate.py +1 -4
generate.py
CHANGED
|
@@ -207,7 +207,4 @@ def generate(
|
|
| 207 |
**model_kwargs,
|
| 208 |
)
|
| 209 |
|
| 210 |
-
return
|
| 211 |
-
"sequences": generated_token_ids,
|
| 212 |
-
"attentions": attention_mask
|
| 213 |
-
}
|
|
|
|
| 207 |
**model_kwargs,
|
| 208 |
)
|
| 209 |
|
| 210 |
+
return generated_token_ids, attention_mask
|
|
|
|
|
|
|
|
|