copali_finetuned / README.md
commissarsilver's picture
Upload folder using huggingface_hub
4b5daef verified
|
raw
history blame
979 Bytes
---
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