File size: 1,160 Bytes
40753b2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
---
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
|