sh train_scripts/causal_qformer.sh is not working!

[leedaehan-kev]

Traceback (most recent call last):
File "train.py", line 16, in
import lavis.tasks as tasks
File "/home/datai/daehan/SEED/SEED_Tokenizer/lavis/init.py", line 16, in
from lavis.models import *
File "/home/datai/daehan/SEED/SEED_Tokenizer/lavis/models/init.py", line 34, in
from lavis.models.blip2_models.blip2 import Blip2Base
File "/home/datai/daehan/SEED/SEED_Tokenizer/lavis/models/blip2_models/blip2.py", line 26, in
from lavis.models.clip_vit import create_clip_vit_L,create_clip_vit_B,create_beit2_tokenizer,create_vit_mage
File "/home/datai/daehan/SEED/SEED_Tokenizer/lavis/models/clip_vit.py", line 15, in
from lavis.models.blip2_models.beit2 import modeling_vqkd
ModuleNotFoundError: No module named 'lavis.models.blip2_models.beit2'

The error "No module named 'lavis.models.blip2_models.beit2'" continuously occurs with BLIP2. If I comment it out, the code doesn't import correctly, leading to concerns about conflicts within the environment. Can you provide a solution?

[laiyingxin2]

same question~