h-zhao1997 / cobra

Cobra: Extending Mamba to Multi-modal Large Language Model for Efficient Inference
MIT License
219 stars 7 forks source link

TypeError: expected Tensor as element 1 in argument 0, but got list #18

Open RoyZry98 opened 1 week ago

RoyZry98 commented 1 week ago

Minor Mistake During Evaluation

A minor mistake occurred during the evaluation process. Here is the error message:

File "./Cobra/cobra/cobra/models/backbones/vision/dinosiglip_vit.py", line 145, in forward
    return torch.cat([dino_patches[0], siglip_patches], dim=2)
TypeError: expected Tensor as element 1 in argument 0, but got list

To fix this issue, the code on line 145 should be updated to:

return torch.cat([dino_patches[0], siglip_patches[0]], dim=2)