search

unum-cloud / uform

Pocket-Sized Multimodal AI for content understanding and generation across multilingual texts, images, and 🔜 video, up to 5x faster than OpenAI CLIP and LLaVA 🖼️ & 🖋️
https://unum-cloud.github.io/uform/
Apache License 2.0
982 stars 56 forks source link

No module named 'uform.models' #79

Closed ake020675 closed 3 months ago

ake020675 commented 3 months ago

the problem is: Traceback (most recent call last): File "/content/ugen-image-captioning-hf/app.py", line 2, in from uform import gen_model File "/usr/local/lib/python3.10/dist-packages/uform/gen_model.py", line 18, in from uform.models import VisualEncoder ModuleNotFoundError: No module named 'uform.models'

Maybe this line should be "from uform.torch_models import VisualEncoder"?

ashvardanian commented 3 months ago

Yes, @ake020675, it should! Can you please open a pull request?

ashvardanian commented 3 months ago

Thank you, @ake020675!

ake020675 commented 3 months ago

Yes, @ake020675, it should! Can you please open a pull request?

I tested the new code and it worked. It's my first time to pull request. Hope i did everything right

ashvardanian commented 3 months ago

Yes, it was great, @ake020675! Don’t hesitate to submit patches to USearch, StringZilla, SimSIMD or other related projects I maintain. Community makes open source better 🤗