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