CoolFace
Modelpublic

user55442/Vilnius-Bus-Stop-LLM

sourceHugging Faceapache-2.0updated 4mo agoView on Hugging Face
0likes7downloads
Model Card

Vilnius Bus Stop LLM

A LoRA adapter fine-tuned on Qwen3-VL-2B-Instruct to recognize Vilnius bus stops in images and describe them in Lithuanian.

Model Details

  • —Model type: Vision-Language Model (LoRA adapter)
  • —Base model: unsloth/Qwen3-VL-2B-Instruct
  • —Language: Lithuanian (lt)
  • —Fine-tuning framework: Unsloth
  • —Task: Image captioning of bus stops in Lithuanian

How to Get Started

python
from peft import PeftModel
from transformers import AutoProcessor, Qwen2VLForConditionalGeneration
import torch

base_model = "unsloth/Qwen3-VL-2B-Instruct"
adapter = "user55442/Vilnius-Bus-Stop-LLM"

model = Qwen2VLForConditionalGeneration.from_pretrained(
    base_model,
    torch_dtype=torch.bfloat16,
    device_map="auto",
    ignore_mismatched_sizes=True
)
model = PeftModel.from_pretrained(model, adapter)
processor = AutoProcessor.from_pretrained(base_model)

Training Details

Dataset

  • —150 daytime images of Vilnius bus stops, captured from varying angles and distances
  • —Captions generated in English by Gemini, then translated to Lithuanian
  • —80/20 train/eval split → 120 training images, 30 test images

Training Procedure

  • —Epochs: 8 (optimal checkpoint ~step 120 before overfitting)
  • —Batch size: 1 with gradient accumulation over 4 steps
  • —Learning rate: 1e-4 (AdamW 8-bit optimizer)
  • —Precision: bfloat16
  • —Image resolution: max 768×768
  • —LoRA target layers: language and attention layers (vision layers frozen)

Evaluation Results

Intrinsic Metrics

MetricBaseFine-tuned
ROUGE-L0.0140.163
Semantic Similarity0.7310.801
BLEU0.33910.130
BERTScore F10.8110.864
Perplexity14.1706544.885

LLM Judge Scores (Gemma-4-31B, scale 1–10)

MetricBaseFine-tuned
Fluency9.775.67
Relevance8.476.37
Factual Accuracy7.605.20
Creativity8.805.40

Limitations

  • —Trained on only 120 images — model shows signs of overfitting after ~120 steps
  • —Perplexity increased sharply (14 → 6544), suggesting the model partially overfit to caption style
  • —LLM judge noted grammatical errors, hallucinations, and incomplete sentences in some outputs
  • —Performance may degrade on bus stops outside Vilnius or in different lighting conditions

Framework Versions

  • —PEFT 0.19.1
  • —Unsloth
  • —Transformers