Views
No views yet
1from peft import PeftModel, PeftConfig
2from transformers import AutoModelForSeq2SeqLM
3
4config = PeftConfig.from_pretrained("ahmedabdelwahed/sft-base-12-epochs")
5model = AutoModelForSeq2SeqLM.from_pretrained("google/mt5-base")
6model = PeftModel.from_pretrained(model, "ahmedabdelwahed/sft-base-12-epochs")