transformers
Optimize ForCausalLMLoss by removing unnecessary contiguous() call to reduce memory overhead
#35646
Merged

Optimize ForCausalLMLoss by removing unnecessary contiguous() call to reduce memory overhead #35646

Rocketknight1 merged 1 commit into huggingface:main from optimize_ForCausalLMLoss
efsotr
efsotr Optimize ForCausalLMLoss by removing unnecessary contiguous() calls t…
0e53a580
efsotr efsotr requested a review from ArthurZucker ArthurZucker 349 days ago
ArthurZucker
ArthurZucker commented on 2025-01-13
efsotr
v-lmn
Rocketknight1
Rocketknight1 approved these changes on 2025-01-16
Rocketknight1
Rocketknight1 Rocketknight1 merged 8ebe9d71 into main 346 days ago
ArthurZucker

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone