rohanch398 comments

Repositories
Issues
Comments

Results 1 comments of


                                            rohanch398

cannot import name 'dot_product_attention_weights' from 'flax.linen.attention

@sanchit-gandhi Looks like problem with windows system. If I install jax==0.4.5 or later, I get "jaxlib" error(No Module found). ModuleNotFoundError: jax requires jaxlib to be installed. If I If I...