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?