nanoVLM
Fine-Tuning PlatformMLOps & Experiment Tracking

nanoVLM

The simplest, fastest repository for training/finetuning small-sized VLMs.

Open Source

About

nanoVLM is the simplest repository for training/finetuning a small sized Vision-Language Model with a lightweight implementation in pure PyTorch. The code itself is very readable and approachable, the model consists of a Vision Backbone (models/vision_transformer.py ~150 lines), Language Decoder (models/language_model.py ~250 lines), Modality Projection (models/modality_projection.py ~50 lines) and the VLM itself (models/vision_language_model.py ~100 lines) and a simple training loop (train.py ~200 lines).

Open Source Health

Not enough history
Stars
5,018
Forks
509
License
Apache-2.0
Last commit
1 years ago
Python

Related Categories