|
--- |
|
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 : |
|
|
|
```python |
|
|
|
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) |
|
|
|
``` |