DAMO-NLP-SG / DAMO-NLP-SG/VideoLLaMA2
QLoRA fin-tunes a custom model with 4-bits, and inference the video, then we got :
- 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