fix: correct batch output list shape when using pipeline

nvtransfer / RULER

This repo contains the source code for RULER: What’s the Real Context Size of Your Long-Context Language Models?

Apache License 2.0

646 stars 43 forks source link

fix: correct batch output list shape when using pipeline #45

Closed Wangmerlyn closed 2 months ago

Wangmerlyn commented 2 months ago

The batch output list shape was incorrect in the pipeline process. This fix ensures the correct shape is applied, improving the overall process stability.

Before fixing: QQ_1722491231511

hsiehjackson commented 2 months ago

Thanks!