Enable device map #30870

darshana1406
darshana1406 added_no_split_modules
b49c3650
darshana1406 added LlavaNextVisionAttention to _no_split_modules
f4bf56ee
amyeroberts
zucchini-nlp
darshana1406
amyeroberts
amyeroberts approved these changes on 2024-05-17
amyeroberts amyeroberts merged 3802e786 into main 1 year ago
darshana1406

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone