takala/financial_phrasebank
Updated • 4.28k • 270
How to use Yasbok/Flan-t5-fine-tune-PEFT-Lora with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("Yasbok/Flan-t5-fine-tune-PEFT-Lora")
model = AutoModelForSeq2SeqLM.from_pretrained("Yasbok/Flan-t5-fine-tune-PEFT-Lora", device_map="auto")