Sayak Paul
58a0c4d58d
[tests] port attention processor tests to use pytest ( #14161 )
...
port attention processor tests to use pytest
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com >
2026-07-21 13:42:34 +05:30
Sayak Paul
273b445426
[tests] fix deprecated attention processor testing. ( #13469 )
...
fix deprecated attention processor testing.
2026-04-14 22:05:39 +05:30
Dhruv Nair
7aa6af1138
[Refactor] Move testing utils out of src ( #12238 )
...
* update
* update
* update
* update
* update
* merge main
* Revert "merge main"
This reverts commit 65efbcead5
2025-08-28 19:53:02 +05:30
Sayak Paul
851dfa30ae
[Tests] Fix more tests sayak ( #10359 )
...
* fixes to tests
* fixture
* fixes
2024-12-23 19:11:21 +05:30
YiYi Xu
e5674015f3
adding back test_conversion_when_using_device_map ( #7704 )
...
* style
* Fix device map nits (#7705 )
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2024-04-18 19:21:32 -10:00
YiYi Xu
a341b536a8
disable test_conversion_when_using_device_map ( #7620 )
...
* disable test
* update
---------
Co-authored-by: yiyixuxu <yixu310@gmail,com>
2024-04-09 09:01:19 -10:00
Will Berman
0bab447670
relax tol attention conversion test ( #3842 )
2023-06-21 12:35:38 -07:00
Will Berman
59aefe9ea6
device map legacy attention block weight conversion ( #3804 )
2023-06-16 10:39:20 -07:00
Will Berman
c6180a311c
add only cross attention to simple attention blocks ( #3011 )
...
* add only cross attention to simple attention blocks
* add test for only_cross_attention re: @patrickvonplaten
* mid_block_only_cross_attention better default
allow mid_block_only_cross_attention to default to
`only_cross_attention` when `only_cross_attention` is given
as a single boolean
2023-04-11 14:38:50 -07:00
