Hi,
I am currently trying to use the OmniMAE model by importing it as such:
from omnimae.omni_mae_model import OmniMAE
However, when I try to use the model, it returns the error:
from vision_transformer import VisionTransformer, Attention, Decoder, PadIm2Video ModuleNotFoundError: No module named 'vision_transformer'
I am not sure if I am importing the model incorrectly or if there is another issue? Thanks!
Hi, I am currently trying to use the OmniMAE model by importing it as such:
from omnimae.omni_mae_model import OmniMAEHowever, when I try to use the model, it returns the error:from vision_transformer import VisionTransformer, Attention, Decoder, PadIm2Video ModuleNotFoundError: No module named 'vision_transformer'I am not sure if I am importing the model incorrectly or if there is another issue? Thanks!