Fix the bug that `Trainer` cannot correctly call `torch_jit_model_eval`#35722

Merged

SunMarc merged 1 commit intohuggingface:mainfrom Wanguy:debug_torch_jit_model_eval

Jan 16, 2025

+3-1

Commits on Jan 16, 2025

Fix the bug that the accelerator.autocast does not pass parameters correctly when calling torch_jit_model_eval (huggingface#35706 )
Wanguy
committed