Views
No views yet
1
2from transformers import AutoModelForCausalLM
3from peft import PeftModel
4
5base_model = AutoModelForCausalLM.from_pretrained("HuggingFaceM4/idefics2-8b")
6peft_model_id = "ayoubkirouane/Idefics2-8b-Finetuned-Lora"
7model = PeftModel.from_pretrained(base_model, peft_model_id)
8