transformers
0c2121f9 - Fix building alibi tensor when num_heads is not a power of 2 (#28380)

Commit

2 years ago

Fix building alibi tensor when num_heads is not a power of 2 (#28380) * Fix building alibi tensor when num_heads is not a power of 2 * Remove print function

References

#28380 - Fix building alibi tensor when num_heads is not a power of 2

Author

abuelnasr0

Parents

53cffeb3

transformers 0c2121f9 - Fix building alibi tensor when num_heads is not a power of 2 (#28380)

transformers
0c2121f9 - Fix building alibi tensor when num_heads is not a power of 2 (#28380)