Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

about use image start and end token #29

Open
Hoteryoung opened this issue Apr 1, 2024 · 2 comments
Open

about use image start and end token #29

Hoteryoung opened this issue Apr 1, 2024 · 2 comments

Comments

@Hoteryoung
Copy link

In the finetune_lora.sh, the argument --mm_use_im_start_end is set to False.
However, based on the paper (see figure below), it should be True.
image
Furthermore, when I change this argument to True.
The following error occurred:

Traceback (most recent call last):
  File "/xxx/Documents/code/geochat/geochat/train/train_mem.py", line 13, in <module>
    train()
  File "/xxx/Documents/code/geochat/geochat/train/train.py", line 952, in train
    model.initialize_vision_tokenizer(model_args, tokenizer=tokenizer)
  File "/xxx/Documents/code/geochat/geochat/model/geochat_arch.py", line 343, in initialize_vision_tokenizer
    embed_tokens_weight = mm_projector_weights["model.embed_tokens.weight"]
KeyError: 'model.embed_tokens.weight'
@Hoteryoung Hoteryoung changed the title about use image start and end toekn about use image start and end token Apr 1, 2024
@732259408
Copy link

@Hoteryoung , do you resolve this issue?

@Hoteryoung
Copy link
Author

Not yet. I just keep it to be False

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants