-
Notifications
You must be signed in to change notification settings - Fork 414
Description
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'