Views
No views yet
google/flan-t5-small (or whichever size you used)eval_loss: 2.59ROUGE-1: 0.10ROUGE-2: 0.03ROUGE-L: 0.1071import torch
2from transformers import T5Tokenizer, T5ForConditionalGeneration
3
4MODEL_PATH = "alakxender/flan-t5-base-alpaca-dv5"
5
6device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
8tokenizer = T5Tokenizer.from_pretrained(MODEL_PATH)
9model = T5ForConditionalGeneration.from_pretrained(MODEL_PATH).to(device)
10
11def generate_response(instruction, input_text):
12 combined_input = f"{instruction.strip()} {input_text.strip()}" if input_text else instruction.strip()
13 inputs = tokenizer(combined_input, return_tensors="pt", truncation=True, max_length=256).to(device)
14
15 output_ids = model.generate(
16 **inputs,
17 max_new_tokens=256,
18 num_beams=8,
19 repetition_penalty=1.5,
20 no_repeat_ngram_size=3,
21 do_sample=True,
22 early_stopping=True,
23 temperature=0.1
24 )
25
26 decoded_output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
27 return decoded_output
28
29# Example usage:
30instruction = "ދީފައިވާ މައުޟޫޢާ ބެހޭގޮތުން ކުރު ޕެރެގްރާފެއް ލިޔެލާށެވެ."
31input_text = "އިއާދަކުރަނިވި ހަކަތަ ބޭނުންކުރުމުގެ މުހިންމުކަން"
32print(generate_response(instruction, input_text))
33
34އިއާދަކުރަނިވި ހަކަތަ ބޭނުންކުރުމުގެ މުހިންމު އެއް މައުޟޫއަކީ ސޯލާ، ވިންޑް، ހައިޑްރޯ، ޖިއޮތަރމަލް، އަދި ހައިޑްރޯއިލެކްޓްރިކް ޕަވަރ ފަދަ އިއާދަކުރަނިވި ހަކަތައިން ގްރީންހައުސް ގޭސްތައް ބޭރުވުން .....
35{
'eval_loss': 1.947984218597412,
'eval_rouge1': 0.19402897739527525,
'eval_rouge2': 0.13442865174751445,
'eval_rougeL': 0.19137712018920253,
'eval_rougeLsum': 0.192215538761906,
'eval_runtime': 46947.7862,
'eval_samples_per_second': 0.11,
'eval_steps_per_second': 0.11,
'epoch': 10.0
}