Views
No views yet
google/flan-t5-large. It was trained specifically for the SciHigh 2026 (Subtask 2) competition to automatically generate concise and accurate scientific titles given research paper abstracts.google/flan-t5-large1import torch
2from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3from peft import PeftModel
4
5# 1. Configuration
6BASE_MODEL_NAME = "google/flan-t5-large"
7ADAPTER_REPO_ID = "Vino1502/scihigh-2026-task2-t5-large"
8
9# 2. Load Tokenizer & Base Model
10tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL_NAME)
11base_model = AutoModelForSeq2SeqLM.from_pretrained(
12 BASE_MODEL_NAME,
13 dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
14 device_map="auto"
15)
16
17# 3. Load LoRA Adapter from your HF Repo
18model = PeftModel.from_pretrained(base_model, ADAPTER_REPO_ID)
19model.eval()
20
21# 4. Sample Inference
22abstract_text = "Your scientific abstract goes here..."
23prompt = f"Generate a concise scientific title for this abstract: {abstract_text}"
24
25# Tokenize and place tensors on model's device
26inputs = tokenizer(prompt, return_tensors="pt", max_length=512, truncation=True).to(model.device)
27
28with torch.no_grad():
29 outputs = model.generate(
30 **inputs,
31 max_length=64,
32 num_beams=2,
33 early_stopping=True
34 )
35
36predicted_title = tokenizer.decode(outputs[0], skip_special_tokens=True)
37print("Predicted Title:", predicted_title)