New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

批量推理时结果异常 #98

Open

ZayIsAllYouNeed opened this issue Dec 5, 2023 · 0 comments

ZayIsAllYouNeed commented Dec 5, 2023

给 LLaMA2-7B 调用 model.generate 函数做推理，"do_sample":False，
如果batch_size=1，则结果很好，
如果batch_size>1，则结果异常，
使用 trainer.predict 批量推理出的结果也是异常的。
#请问大佬们遇到过这种情况吗？

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment