How to use code-world/google-flan-t5-small-lora-ft with PEFT:
from peft import PeftModel from transformers import AutoModelForSeq2SeqLM base_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small") model = PeftModel.from_pretrained(base_model, "code-world/google-flan-t5-small-lora-ft")
How to use code-world/google-flan-t5-small-lora-ft with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("code-world/google-flan-t5-small-lora-ft", device_map="auto")
