DAMO-NLP-SG / DAMO-NLP-SG/VideoLLaMA2

QLoRA fin-tunes a custom model with 4-bits, and inference the video, then we got :

Open
#106 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
1.3k
Forks
90
PR merge metrics
No merged PRs in 30d

Description

RuntimeError: Error(s) in loading state_dict for Videollama2MistralForCausalLM:
size mismatch for model.mm_projector.readout.0.weight: copying a param with shape torch.Size([4096, 4096]) from checkpoint, the shape in current model is torch.Size([8388608, 1]).
size mismatch for model.mm_projector.readout.2.weight: copying a param with shape torch.Size([4096, 4096]) from checkpoint, the shape in current model is torch.Size([8388608, 1]).

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reproducing the QLoRA 4-bit custom-model load and video inference path described in the issue. Compare the checkpoint and current model shapes for model.mm_projector.readout.0.weight and readout.2.weight; done means the fine-tuned checkpoint loads without size-mismatch errors and inference completes.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.