transformers
fixed the issue of DPO trainer that using one node and mutiple GPUs and set the device_map='auto'
#29695

Merged

fixed the issue of DPO trainer that using one node and mutiple GPUs and set the device_map='auto' #29695

amyeroberts merged 7 commits into huggingface:main from pengwei715:feature/fix_dpo_los_update

fixed the issue of DPO trainer that using one node and mutiple GPUs

7a17fea9

amyeroberts commented on 2024-03-18

younesbelkada commented on 2024-03-18

before update, add the assert

91b729c3

run the ruff formatter

4e2f65b6

Merge branch 'main' into feature/fix_dpo_los_update

7d4d6126

guy1992l commented on 2024-03-18

younesbelkada commented on 2024-03-19

younesbelkada requested a review from

muellerzr 1 year ago

younesbelkada requested a review from

amyeroberts 1 year ago

Update src/transformers/trainer.py

155ad13f

muellerzr approved these changes on 2024-03-19

remember to do make style and make quality before commit

acae705b

amyeroberts approved these changes on 2024-03-19

Update src/transformers/trainer.py

3ca37964

pengwei715 requested a review from

amyeroberts 1 year ago

amyeroberts approved these changes on 2024-03-20

amyeroberts merged 8692aa88 into main 1 year ago

Reviewers

amyeroberts

muellerzr

younesbelkada

guy1992l

Assignees

No one assigned

Labels

None yet

Milestone

No milestone