transformers
2f121eaf - Fix: Update optimization.py (#44909)

Commit
40 days ago
Fix: Update optimization.py (#44909) * Update optimization.py The `get_scheduler` function was identifying the `inverse_sqrt` scheduler type but failing to pass `**scheduler_specific_kwargs` to the underlying `get_inverse_sqrt_schedule` function. This caused user-defined parameters like `timescale` to be silently ignored. This commit adds the missing kwargs to the function call at line 664. Fixes #44908 * Update optimization.py * Update optimization.py * Update optimization.py
Author
Parents
Loading