transformers
Replace default split function with jnp.split() in flax models
#37001

Merged

Replace default split function with jnp.split() in flax models #37001

Rocketknight1 merged 1 commit into huggingface:main from premmurugan229:fix_split_function_in_flax_models

github-actions marked this pull request as draft 339 days ago

Rocketknight1 added Flax

premmurugan229 force pushed from 0fa9611a to d1fe5523 339 days ago

premmurugan229 marked this pull request as ready for review 339 days ago

github-actions requested a review from

ArthurZucker 339 days ago

Replace split with jnp's split function for flax models (#36854)

4ab4050e

Rocketknight1 force pushed from d1fe5523 to 4ab4050e 338 days ago

Rocketknight1 approved these changes on 2025-03-27

Rocketknight1 merged 4cc65e99 into main 338 days ago

Reviewers

Rocketknight1

ArthurZucker

Assignees

No one assigned

Labels

Flax

Milestone

No milestone