Views
No views yet
GanjinZero/biobart-v2-large (~406M parameters)1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2from peft import PeftModel
3
4base = "GanjinZero/biobart-v2-large"
5adapter = "priyanshi-garg/biobart-large-lora" # update with actual HF repo name
6
7tokenizer = AutoTokenizer.from_pretrained(base)
8model = AutoModelForSeq2SeqLM.from_pretrained(base, torch_dtype="auto")
9model = PeftModel.from_pretrained(model, adapter)
10model = model.merge_and_unload() # merge LoRA weights for inference
11
12inputs = tokenizer("Patient clinical note here...", return_tensors="pt", truncation=True, max_length=1024)
13summary_ids = model.generate(**inputs, max_new_tokens=256, num_beams=4)
14print(tokenizer.decode(summary_ids[0], skip_special_tokens=True))