huggingface / optimum-nvidia

Apache License 2.0
872 stars 86 forks source link

Fix issue returning only the first batch item after pipeline call. #29

Closed mfuntowicz closed 9 months ago

mfuntowicz commented 9 months ago

For some reason, maybe a debugging artifact, only [0] was being used during the postprocess phase resulting in a single batch output.

Fixes #28