[BUG] VisionEmbedding readme example error

[evelynmitchell]

In the VisionEmbedding example, run in a colab notebook:

ImportError                               Traceback (most recent call last)
[<ipython-input-10-de694ebd704c>](https://localhost:8080/#) in <cell line: 2>()
      1 import torch
----> 2 from zeta.nn import VisionEmbedding
      3 
      4 # Create an instance of VisionEmbedding
      5 vision_embedding = VisionEmbedding(

ImportError: cannot import name 'VisionEmbedding' from 'zeta.nn' (/usr/local/lib/python3.10/dist-packages/zeta/nn/__init__.py)

zeta/nn/init.py imports

from zeta.nn.embeddings import * 

So, I thought the fix was to change the example to:

from zeta.nn.embeddings import VisionEmbedding

but that doesn't work:
It appears VisionEmbedding isn't exported in zeta.nn.embeddings init,py

Upvote & Fund

• We're using Polar.sh so you can upvote and help fund this issue.
• We receive the funding once the issue is completed & confirmed by you.
• Thank you in advance for helping prioritize & fund our backlog.

[Kye Gomez]

@evelynmitchell i think I fixed it lmk

[evelynmitchell]

Looks good! Thanks