Views
No views yet
1"Di seguito è riportata un'istruzione che descrive un'attività, abbinata ad un input che fornisce ulteriore informazione. " \
2"Scrivi una risposta che soddisfi adeguatamente la richiesta.\n\n" \
3f"### Istruzione:\n{instruction}\n\n### Input:\n{input}\n\n### Risposta:\n{response}"1"Di seguito è riportata un'istruzione che descrive un'attività. " \
2"Scrivi una risposta che soddisfi adeguatamente la richiesta.\n\n" \
3f"### Istruzione:\n{instruction}\n\n### Risposta:\n{response}"1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "swap-uniba/LLaMAntino-2-7b-hf-dolly-ITA"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(model_id)
7
8instruction_text = "Estrai i nomi propri di persona dal testo che segue"
9input_text = "Marco ha incontrato Matteo per strada e hanno parlato di Mirco"
10
11prompt = "Di seguito è riportata un'istruzione che descrive un'attività, accompagnata da un input che aggiunge ulteriore informazione. " \
12 f"Scrivi una risposta che completi adeguatamente la richiesta.\n\n" \
13 f"### Istruzione:\n{instruction_text}\n\n" \
14 f"### Input:\n{input_text}\n\n" \
15 f"### Risposta:\n"
16
17input_ids = tokenizer(prompt, return_tensors="pt").input_ids
18outputs = model.generate(input_ids=input_ids)
19
20print(tokenizer.batch_decode(outputs.detach().cpu().numpy()[:, input_ids.shape[1]:], skip_special_tokens=True)[0])model = AutoModelForCausalLM.from_pretrained(model_id, load_in_8bit=True)1@misc{basile2023llamantino,
2 title={LLaMAntino: LLaMA 2 Models for Effective Text Generation in Italian Language},
3 author={Pierpaolo Basile and Elio Musacchio and Marco Polignano and Lucia Siciliani and Giuseppe Fiameni and Giovanni Semeraro},
4 year={2023},
5 eprint={2312.09993},
6 archivePrefix={arXiv},
7 primaryClass={cs.CL}
8}