AILab-CVC / SEED

Official implementation of SEED-LLaMA (ICLR 2024).
https://ailab-cvc.github.io/seed
Other
576 stars 31 forks source link

train_scripts/causal_qformer.sh not working #30

Open shubhamgarg21 opened 8 months ago

shubhamgarg21 commented 8 months ago

Hi,

While trying to run step 3 of Training SEED Tokenization, I am observing following error:

Traceback (most recent call last):
  File "/SEED/SEED_Tokenizer/train.py", line 16, in <module>
    import lavis.tasks as tasks
  File "/SEED/SEED_Tokenizer/lavis/__init__.py", line 15, in <module>
    from lavis.datasets.builders import *
  File "/SEED/SEED_Tokenizer/lavis/datasets/builders/__init__.py", line 15, in <module>
    from lavis.datasets.builders.image_text_pair_builder import (
  File "/SEED/SEED_Tokenizer/lavis/datasets/builders/image_text_pair_builder.py", line 16, in <module>
    from lavis.datasets.datasets.mmc4_dataset import MMC4Dataset
ModuleNotFoundError: No module named 'lavis.datasets.datasets.mmc4_dataset'
geyuying commented 8 months ago

You need to comment out the dataset builder that is not provided in this repo.