---
base_model: OpenGVLab/InternVL2-2B
library_name: peft
pipeline_tag: image-text-to-text
tags:
- lora
- internvl2
- vision-language
- android-control
license: apache-2.0
---

# 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](https://huggingface.co/OpenGVLab/InternVL2-2B)
- **LoRA Rank**: 8
- **LoRA Alpha**: 32
- **Training Rounds**: 5 (global_lora_5)

## 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