Unverified Commit ed98ad35 authored by Pavel Gein's avatar Pavel Gein Committed by GitHub
Browse files

Fix usage of unpad_input function (#35925)

Fix usage of unpad_function

See https://github.com/huggingface/transformers/issues/35899

In the [commit](https://github.com/Dao-AILab/flash-attention/commit/cdbbe844b1c0bcba3362e1f8c8af4d6f6d0bf300

) return type of `unpad_input` was changed.
Now the code support older and newer versions

Co-authored-by: default avatarPavel Gein <pavel.gein@gmail.com>
parent 7aee036e
Showing with 1 addition and 1 deletion
+1 -1
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment