|
|
--- |
|
|
datasets: |
|
|
- ayoubkirouane/llava-instruct-small |
|
|
library_name: peft |
|
|
pipeline_tag: image-text-to-text |
|
|
language: |
|
|
- en |
|
|
--- |
|
|
|
|
|
|
|
|
## Base model : |
|
|
- xtuner/llava-phi-3-mini-hf |
|
|
|
|
|
## Dataset : |
|
|
- ayoubkirouane/llava-instruct-small |
|
|
|
|
|
|
|
|
## Get started : |
|
|
|
|
|
```python |
|
|
|
|
|
from transformers import AutoModelForCausalLM |
|
|
from peft import PeftModel |
|
|
|
|
|
base_model = AutoModelForCausalLM.from_pretrained("xtuner/llava-phi-3-mini-hf") |
|
|
peft_model_id = "ayoubkirouane/llava-phi3-instruct-Lora" |
|
|
model = PeftModel.from_pretrained(base_model, peft_model_id) |
|
|
|
|
|
``` |
