Spaces:

sanchit-gandhi
/

whisper-jax-spaces

Running on TPU v5e

update layers.py for JAX deprecate "shape"

by zxyse - opened Aug 8, 2024

←

zxyse

Aug 8, 2024

used generic tuple instead of jax.core.NamedShape
used np to get total size of shape

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Ready to merge

This branch is ready to get merged automatically.

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment