PKU-YuanGroup / Chat-UniVi

[CVPR 2024 Highlight🔥] Chat-UniVi: Unified Visual Representation Empowers Large Language Models with Image and Video Understanding
https://arxiv.org/abs/2311.08046
Apache License 2.0
755 stars 41 forks source link

16GB-VRAM run Chat-UniVi-7B-v1.5 model? #37

Open joseph16388 opened 4 months ago

joseph16388 commented 4 months ago

Can 16GB-VRAM run Chat-UniVi-7B-v1.5 model? thanks

jpthu17 commented 3 months ago

16GB-VRAM doesn't seem like enough for training the model, you can try using Lora and reducing the batch size.

For model inference, 16GB-VRAM is sufficient.