Update generate.py

generate.py

@@ -207,4 +207,7 @@ def generate(
         **model_kwargs,
     )
 
-    return generated_token_ids, attention_mask
+    return {
+        "sequences": generated_token_ids,
+        "attentions": attention_mask
+    }