Modelpublic

bmh201708/internvl2-2b-lora-categories

sourceHugging Faceapache-2.0updated 8mo agoView on Hugging Face

1likes

Model Card

InternVL2-2B LoRAs for Android Control Categories

This repository contains LoRA adapters for InternVL2-2B trained on different Android UI control task categories.

Available Categories

Category	Subdirectory
Shopping	`shopping/`
Traveling	`traveling/`
Office	`office/`
Lives	`lives/`
Entertainment	`entertainment/`

Model Details

—Base Model: OpenGVLab/InternVL2-2B
—LoRA Rank: 8
—LoRA Alpha: 32
—Training Rounds: 5 (globallora5)

Usage

python

from transformers import AutoModel
from peft import PeftModel

base_model = AutoModel.from_pretrained("OpenGVLab/InternVL2-2B", trust_remote_code=True)

# Load specific category LoRA
category = "shopping"  # or: traveling, office, lives, entertainment
model = PeftModel.from_pretrained(base_model, "bmh201708/internvl2-2b-lora-categories", subfolder=category)

License

Apache 2.0