copali_finetuned / README.md
commissarsilver's picture
Upload folder using huggingface_hub
4b5daef verified
|
raw
history blame
979 Bytes
metadata
base_model: vidore/colpaligemma-3b-mix-448-base
library_name: peft

Model Card for Model ID

Model Details

Model Description

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

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