transformers
Fix GPU OOM for `mistral.py::Mask4DTestHard`
#31212
Merged

Fix GPU OOM for `mistral.py::Mask4DTestHard` #31212

ydshieh merged 4 commits into main from fix_Mask4DTestHard_for_mistral
ydshieh
ydshieh build
4c5082b8
ydshieh build
d4ce9710
ydshieh build
e7b48d8f
ydshieh
ydshieh commented on 2024-06-03
ydshieh ydshieh requested a review from amyeroberts amyeroberts 1 year ago
ydshieh
ydshieh commented on 2024-06-03
HuggingFaceDocBuilderDev
amyeroberts
amyeroberts approved these changes on 2024-06-03
ydshieh build
db725e91
ydshieh ydshieh merged 8a1a23ae into main 1 year ago
ydshieh ydshieh deleted the fix_Mask4DTestHard_for_mistral branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone