ayoubkirouane's picture
Update README.md
2239c13 verified
|
raw
history blame
526 Bytes
metadata
library_name: transformers
datasets:
  - nielsr/docvqa_1200_examples
language:
  - en
pipeline_tag: image-text-to-text

Base Model :

  • HuggingFaceM4/idefics2-8b

Dataset :

  • nielsr/docvqa_1200_examples

Get started :


from transformers import AutoModelForCausalLM
from peft import PeftModel

base_model = AutoModelForCausalLM.from_pretrained("HuggingFaceM4/idefics2-8b")
peft_model_id = "ayoubkirouane/Idefics2-8b-Finetuned-Lora"
model = PeftModel.from_pretrained(base_model, peft_model_id)