Views
No views yet
| Component | Details |
|---|---|
| Base LLM | Qwen/Qwen2.5-Coder-7B-Instruct (7.62B params) |
| Vision Encoder | CLIP ViT-L/14 (frozen, 304M params) |
| LoRA Adapters | r=64, alpha=128 (161.5M trainable params) |
| Fusion | VisionLanguageFusion with text_gate (16.8M params) |
| Total | 8.1B params, 182.5M trainable (2.25%) |
checkpoints/
├── phase3_final/ ← Best checkpoint for inference
│ ├── lora/ ← LoRA adapter weights
│ ├── vision/ ← Vision projection weights
│ └── fusion/ ← Fusion layer weights
├── phase3_all_step2500_final/
├── phase3_all_step2000/
├── phase3_all_step1500/
└── ... (earlier phases)1from src.model.mindi_model import MINDI15
2import torch
3
4model = MINDI15(
5 model_name="Qwen/Qwen2.5-Coder-7B-Instruct",
6 clip_model="openai/clip-vit-large-patch14",
7 hidden_size=3584,
8 num_visual_tokens=256,
9 torch_dtype=torch.bfloat16,
10)
11model.load("checkpoints/phase3_final")
12model.eval()
13
14response = model.generate(
15 prompt="Build a Next.js landing page",
16 max_new_tokens=2048,
17 temperature=0.7,
18)