transformers
3611fc90 - compute_loss in trainer failing to label shift for PEFT model when label smoothing enabled. (#25044)

Commit
2 years ago
compute_loss in trainer failing to label shift for PEFT model when label smoothing enabled. (#25044) * added PeftModelForCausalLM to MODEL_FOR_CAUSAL_LM_MAPPING_NAMES dict * check for PEFT model in compute_loss section --------- Co-authored-by: Nathan Brake <nbrake3@mmm.com>
Author
Parents
Loading