base_model: vidore/colpaligemma-3b-mix-448-base | |
library_name: peft | |
# Model Card for Model ID | |
<!-- Provide a quick summary of what the model is/does. --> | |
## Model Details | |
### Model Description | |
<!-- Provide a longer summary of what this model is. --> | |
- **Model type:** PaliGemma | |
- **Finetuned from model [optional]:** vidore/colpali-v1.2 | |
## Uses | |
This model is finetuned from `vidore/colpali-v1.2` using the PEFT library. To use this model, you can use the following code: | |
```python | |
from transformers import AutoModelForCausalLM, AutoTokenizer | |
from peft import PeftModel | |
### This will load the basemodel ### | |
# Load the base model | |
model_name = "colpali_finetuned" | |
model = AutoModelForCausalLM.from_pretrained(model_name) | |
# Load the tokenizer | |
tokenizer = AutoTokenizer.from_pretrained(model_name) | |
### This will load the adapter model ### | |
peft_model_path='colpali_finetuned/checkpoint-587' | |
model = PeftModel.from_pretrained(peft_model_path) | |
``` | |
- PEFT 0.11.1 |