transformers

7e1eff76 - Allow scheduler parameters (#26480)

Allow scheduler parameters (#26480) * Allow for scheduler kwargs * Formatting * Arguments checks, passing the tests * Black failed somehow --------- Co-authored-by: Pierre <pierre@avatarin.com>