transformers
fix(trainer): Avoid moving model with device_map
#41032
Merged

fix(trainer): Avoid moving model with device_map #41032

The5cheduler
google-labs-jules[bot] fix(trainer): Avoid moving model with device_map
d345dcce
Rocketknight1
SunMarc
SunMarc approved these changes on 2025-09-24
The5cheduler Added the logger warning for the move model
61e1e13d
The5cheduler
The5cheduler Merge branch 'huggingface:main' into fix-trainer-offload-runtime-error
b8c39694
SunMarc SunMarc enabled auto-merge (squash) 237 days ago
SunMarc SunMarc merged de9a75f5 into main 237 days ago
HuggingFaceDocBuilderDev

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone