How to use ravi-huggingface/test_dir with PEFT:
from peft import PeftModel from transformers import AutoModelForSeq2SeqLM base_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base") model = PeftModel.from_pretrained(base_model, "ravi-huggingface/test_dir")
How to use ravi-huggingface/test_dir with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("ravi-huggingface/test_dir", device_map="auto")
