Forgot to have the outputs named text instead of prompts

397714f5 · rprenger · 4b11fd9d · 397714f5
Commit 397714f5 authored Oct 07, 2021 by rprenger
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

megatron/text_generation_server.py megatron/text_generation_server.py +1 -1

No files found.
--- a/megatron/text_generation_server.py
+++ b/megatron/text_generation_server.py
@@ -107,7 +107,7 @@ class MegatronGenerate(Resource):
                                                                 top_p,
                                                                 add_BOS) 
-        return jsonify({"prompts": response,
+        return jsonify({"text": response,
            "segments": response_seg,
            "logprobs": response_logprobs})