Views
No views yet

1from transformers import AutoTokenizer, pipeline
2import torch
3
4model = "rhaymison/gemma-portuguese-tom-cat-2b-it"
5
6tokenizer = AutoTokenizer.from_pretrained(model)
7pipeline = pipeline(
8 "text-generation",
9 model=model,
10 model_kwargs={"torch_dtype": torch.bfloat16},
11 device="cuda",
12)
13
14messages = [
15 {
16 "role": "system",
17 "content": "Abaixo está uma instrução que descreve uma tarefa, juntamente com uma entrada que fornece mais contexto. Escreva uma resposta que complete adequadamente o pedido."
18 },
19 {"role": "user", "content": "Me conte sobre a ida do homem a Lua."},
20]
21prompt = pipeline.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
22outputs = pipeline(
23 prompt,
24 max_new_tokens=256,
25 do_sample=True,
26 temperature=0.2,
27 top_k=50,
28 top_p=0.95
29)1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer2 = AutoTokenizer.from_pretrained("rhaymison/gemma-portuguese-tom-cat-2b-it")
4model2 = AutoModelForCausalLM.from_pretrained("rhaymison/gemma-portuguese-tom-cat-2b-it", device_map={"":0})
5tokenizer2.pad_token = tokenizer2.eos_token
6tokenizer2.add_eos_token = True
7tokenizer2.add_bos_token, tokenizer2.add_eos_token
8tokenizer2.padding_side = "right"1def format_template( question:str):
2 system_prompt = "Abaixo está uma instrução que descreve uma tarefa, juntamente com uma entrada que fornece mais contexto. Escreva uma resposta que complete adequadamente o pedido."
3 text = f"""<bos>system
4 {system_prompt}<end_of_turn>
5 <start_of_turn>user
6 ###instrução: {question} <end_of_turn>
7 <start_of_turn>model"""
8 return text
9
10question = format_template("Me conte sobre a ida do homem a Lua")
11
12device = "cuda:0"
13
14inputs = tokenizer2(text, return_tensors="pt").to(device)
15
16outputs = model2.generate(**inputs, max_new_tokens=256, do_sample=False)
17
18output = tokenizer2.decode(outputs[0], skip_special_tokens=True, skip_prompt=True)
19print(output.replace("model"," "))| Metric | Value |
|---|---|
| Average | 31.76 |
| ENEM Challenge (No Images) | 27.71 |
| BLUEX (No Images) | 29.07 |
| OAB Exams | 27.97 |
| Assin2 RTE | 46.84 |
| Assin2 STS | 14.06 |
| FaQuAD NLI | 29.39 |
| HateBR Binary | 46.59 |
| PT Hate Speech Binary | 45.36 |
| tweetSentBR | 18.86 |