ayoubkirouane's picture
Update README.md
cc45ef8 verified
metadata
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 :


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)