transformers
7ebfd4b6 - Replace all instances of jnp.DeviceArray with jax.Array

Commit

2 years ago

Replace all instances of jnp.DeviceArray with jax.Array

References

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

Author

mariecwhite

mariecwhite

Committer

mariecwhite

mariecwhite

Parents

Loading