Error when initializing tokenizer

SHTUPLUS / Pix2Grp_CVPR2024

BSD 3-Clause "New" or "Revised" License

30 stars 2 forks source link

Error when initializing tokenizer #10

Open LinxinS97 opened 2 months ago

LinxinS97 commented 2 months ago

In lavis/models/blip_models/blip_rel_det.py

tokenizer = StageBertTokenizer.from_pretrained(
            "/public/home/lirj2/projects/LAVIS_GITM/data/bert-base-uncased",
            local_files_only=True)

I met a hardcoded path for Bert in the code, which I cannot find in your instructions. Could you provide this weight (or reference) in your instructions?

Scarecrow0 commented 2 months ago

Sorry for this missing hardcoded URL; this is the standard pretrain weight for the BERT model. You can download manually from: https://huggingface.co/google-bert/bert-base-uncased or change the code as: StageBertTokenizer.from_pretrained("bert-base-uncased"). The former is faster.