Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

update transformers, module 'jax.core' has no attribute 'NamedShape' error #206

Open
devops724 opened this issue Jan 15, 2025 · 0 comments

Comments

@devops724
Copy link

in upgraded transformers
whisper-jax 0.0.1 requires transformers<4.35.0,>=4.27.4, but you have transformers 4.48.0 which is incompatible.

i get this error
from whisper_jax import FlaxWhisperPipline
File "/home/user/miniconda3/lib/python3.12/site-packages/whisper_jax/init.py", line 18, in
from .modeling_flax_whisper import FlaxWhisperForConditionalGeneration
File "/home/user/miniconda3/lib/python3.12/site-packages/whisper_jax/modeling_flax_whisper.py", line 57, in
from whisper_jax import layers
File "/home/user/miniconda3/lib/python3.12/site-packages/whisper_jax/layers.py", line 63, in
def _compute_fans(shape: jax.core.NamedShape, in_axis=-2, out_axis=-1):
^^^^^^^^^^^^^^^^^^^
File "/home/user/miniconda3/lib/python3.12/site-packages/jax/_src/deprecations.py", line 57, in getattr
raise AttributeError(f"module {module!r} has no attribute {name!r}")
AttributeError: module 'jax.core' has no attribute 'NamedShape'

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant