transformers
Allow `mlm_probability` to be set to `None` when `mlm=False` in DataCollatorForLanguageModeling (#38522)
#38537
Merged

Allow `mlm_probability` to be set to `None` when `mlm=False` in DataCollatorForLanguageModeling (#38522) #38537

KameniAlexNea
Rocketknight1
Rocketknight1 approved these changes on 2025-06-03
KameniAlexNea
Rocketknight1
mlm_probability in DataCollatorForLanguageModeling should be validate…
b5c520f2
Change mlm_probability to Optional in DataCollatorForLanguageModeling…
e17603a4
Rocketknight1 Rocketknight1 force pushed from 4e2e4161 to e17603a4 219 days ago
Rocketknight1
Rocketknight1 Rocketknight1 enabled auto-merge (squash) 219 days ago
HuggingFaceDocBuilderDev
Rocketknight1 Rocketknight1 merged 8f630651 into main 218 days ago
KameniAlexNea KameniAlexNea deleted the 38522-mlm-datacollator-flm branch 218 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone