Skip to content

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

@devops724

Description

@devops724

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'

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions