Views
No views yet
google/flan-t5-baseknkarthick/dialogsum1import torch
2from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3from peft import PeftModel
4
5base_id = "google/flan-t5-base"
6adapter_id = "prithvi1029/flan-t5-dialogsum-lora"
7
8tok = AutoTokenizer.from_pretrained(base_id)
9model = AutoModelForSeq2SeqLM.from_pretrained(
10 base_id, device_map='auto', torch_dtype=torch.float16
11)
12model = PeftModel.from_pretrained(model, adapter_id)
13
14dialogue = "A: Hey, are you free tomorrow?\nB: Yes, what’s up?\nA: Need help with a project."
15prompt = f"Summarize the following conversation.\n\n{dialogue}\n\nSummary:"
16inputs = tok(prompt, return_tensors='pt').to(model.device)
17out = model.generate(**inputs, max_new_tokens=128)
18print(tok.decode(out[0], skip_special_tokens=True))