This commit is contained in:
somebody
2022-09-22 22:05:48 -05:00
parent ca356d4d6f
commit 0b4ce13eb8

View File

@@ -4974,7 +4974,7 @@ class GenerationResult:
):
# Shave prompt off of encoded response. Decoded does not return prompt.
# TODO: Does MTJ generation shave this off automatically? Test it!
__debug("shape", out_batches.shape)
print("shape", out_batches.shape)
if output_includes_prompt:
self.encoded = out_batches[:, len(prompt) - 1:]