# Description | |
A Blip-2 classifier fine-tuned for classification tasks on the EasyVQA dataset. See the [github repository](https://github.com/atomwalk12/VisualQA/). | |
If you decide to reuse this model, here are the LoRa and bnb configurations that were used for training: | |
```python | |
BitsAndBytesConfig( | |
load_in_4bit=True, | |
bnb_4bit_quant_type="nf4", | |
bnb_4bit_compute_dtype=torch.float32, | |
) | |
LoraConfig( | |
r=8, | |
lora_alpha=8, | |
lora_dropout=0.1, | |
target_modules="all-linear", | |
init_lora_weights="gaussian", | |
) | |
``` |