NanbeigeVLM
Collection
Vision gained Nanbeige4.1-3B Models • 3 items • Updated
How to use SkyAsl/Nanbeige4.1-VLM-LoRA with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("Nanbeige/Nanbeige4.1-3B")
model = PeftModel.from_pretrained(base_model, "SkyAsl/Nanbeige4.1-VLM-LoRA")LoRA adapter weights from Stage 2 instruction fine-tuning. Use this if you want the smallest download or want to merge manually.
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
import torch
base = AutoModelForCausalLM.from_pretrained(
"Nanbeige/Nanbeige4.1-3B",
trust_remote_code=True,
torch_dtype=torch.bfloat16,
)
model = PeftModel.from_pretrained(base, "SkyAsl/Nanbeige4.1-VLM-Stage2-LoRA")
# For full VLM inference, use the merged version instead:
# SkyAsl/Nanbeige4.1-VLM-Stage2
Base model
Nanbeige/Nanbeige4-3B-Base