使用 whisper-jax 时,报错:
AttributeError: module 'jax.core' has no attribute 'NamedShape'
根据 isssue : https://github.com/sanchit-gandhi/whisper-jax/issues/199
Changing NamedShape to DShapedArray in “whisper_jax/layers.py”, line 63 gets it working.
去 site-packages 中的 whisper_jax 库,将 layers.py
中 63 行的 NamedShape
改为 DShapedArray
,处理了这个问题
2024-11-11(一)