eleuther generate until

This commit is contained in:
Xi Yan 2024-10-08 23:57:22 -07:00
parent 6abef716dd
commit 9c38d9ae13
3 changed files with 34 additions and 40 deletions

View file

@ -102,9 +102,8 @@ class EleutherEvalsWrapper(LM):
stream=False,
)
print(response)
res.append(response.completion_message)
res.append(response.completion_message.content)
print(response)
return res