PR #24875 Remove jnp.DeviceArray since it is deprecated.

Remove jnp.DeviceArray since it is deprecated. #24875

sanchit-gandhi merged 3 commits into huggingface:main from mariecwhite:mariecwhite-remove-jnp-devicearray

sanchit-gandhi commented on 2023-07-25

mariecwhite force pushed 2 years ago

ayaka14732 requested changes on 2023-08-02

mariecwhite force pushed 2 years ago

ayaka14732 approved these changes on 2023-08-02

Remove jnp.DeviceArray since it is deprecated.

39394260

Replace all instances of jnp.DeviceArray with jax.Array

7ebfd4b6

Update src/transformers/models/bert/modeling_flax_bert.py

0a7c2896

mariecwhite force pushed to 0a7c2896 2 years ago

ayaka14732 approved these changes on 2023-08-04

sgugger approved these changes on 2023-08-04

sanchit-gandhi approved these changes on 2023-08-04

sanchit-gandhi merged a6e6b1c6 into main 2 years ago

Reviewers

sanchit-gandhi

sgugger

ayaka14732

Assignees

No one assigned

Labels

None yet

Milestone

No milestone