File size: 841 Bytes
c165322 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
---
language: en
license: mit
library_name: nanovlm
tags:
- vision-language
- multimodal
- smollm2
- siglip
---
# nanoVLM - devilops/nanoVLM-222M
This is a nano Vision-Language Model (nanoVLM) trained as part of the COM-304 course.
## Model Description
The model consists of three main components:
- **Vision Backbone**: Pretrained `google/siglip-base-patch16-224`
- **Language Model**: Pretrained `HuggingFaceTB/SmolLM2-135M`
- **Modality Projector**: A learnable linear layer with Pixel Shuffle reduction.
## Usage
You can load this model using the `VisionLanguageModel` class from the `nanovlm` repository.
```python
from models.vision_language_model import VisionLanguageModel
import torch
device = "cuda" if torch.cuda.is_available() else "cpu"
model = VisionLanguageModel.from_pretrained("devilops/nanoVLM-222M").to(device)
```
|