Skip to content

Conversation

@yiyixuxu
Copy link
Collaborator

@yiyixuxu yiyixuxu commented Apr 9, 2024

disable test_conversion_when_using_device_map test for now


class DeprecatedAttentionBlockTests(unittest.TestCase):
def test_conversion_when_using_device_map(self):
# To-DO for Sayak: enable this test again and to test `device_map='balanced'` once we have this in accelerate https://github.com/huggingface/accelerate/pull/2641
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

cc @sayakpaul here

@yiyixuxu yiyixuxu merged commit a341b53 into main Apr 9, 2024
@sayakpaul sayakpaul deleted the comment-off-device-depre-attn-test branch April 10, 2024 02:45
sayakpaul pushed a commit that referenced this pull request Dec 23, 2024
* disable test

* update

---------

Co-authored-by: yiyixuxu <yixu310@gmail,com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants