Views
No views yet

1from transformers import AutoTokenizer, pipeline
2import torch
3
4model = "rhaymison/gemma-portuguese-luana-2b"
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)
30print(outputs[0]["generated_text"][len(prompt):].replace("model",""))
31
32#A viagem à Lua foi um esforço monumental realizado pela Agência Espacial dos EUA entre 1969 e 1972.
33#Foi um marco significativo na exploração espacial e na ciência humana.
34#Aqui está uma visão geral de sua jornada: 1. O primeiro voo espacial humano foi o de Yuri Gagarin, que voou a Terra em 12 de abril de 1961.1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer2 = AutoTokenizer.from_pretrained("rhaymison/gemma-portuguese-luana-2b")
4model2 = AutoModelForCausalLM.from_pretrained("rhaymison/gemma-portuguese-luana-2b", 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"1text = f"""<start_of_turn>user
2Abaixo 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###instrução:Me conte sobre a ida do homem a Lua.<end_of_turn>
4<start_of_turn>model """
5
6device = "cuda:0"
7
8inputs = tokenizer2(text, return_tensors="pt").to(device)
9
10outputs = model2.generate(**inputs, max_new_tokens=256, do_sample=False)
11
12output = tokenizer2.decode(outputs[0], skip_special_tokens=True, skip_prompt=True)
13print(output.replace("model"," "))
14
15#A viagem à Lua foi um esforço monumental realizado pela Agência Espacial dos EUA entre 1969 e 1972.
16#Foi um marco significativo na exploração espacial e na ciência humana.
17#Aqui está uma visão geral de sua jornada: 1. O primeiro voo espacial humano foi o de Yuri Gagarin, que voou a Terra em 12 de abril de 1961. 1text = f"""<start_of_turn>user
2Abaixo 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###instrução:Me explique como funciona um computador.<end_of_turn>
4<start_of_turn>model """
5
6device = "cuda:0"
7
8inputs = tokenizer2(text, return_tensors="pt").to(device)
9
10outputs = model2.generate(**inputs, max_new_tokens=256, do_sample=False)
11
12output = tokenizer2.decode(outputs[0], skip_special_tokens=True, skip_prompt=True)
13print(output.replace("model"," "))
14
15#Um computador é um dispositivo eletrônico que pode executar tarefas que um humano pode fazer.
16#Ele usa um conjunto de circuitos elétricos, componentes eletrônicos e software para processar informações e executar tarefas.
17#Os componentes de um computador incluem um processador, memória, unidade de armazenamento, unidade de processamento gráfica,
18#unidade de controle, unidade de entrada e saída,e dispositivos de entrada e saída.
19#O processador é o coração do computador e executa instruções de software.A memória é onde o computador armazena| Metric | Value |
|---|---|
| Average | 34.94 |
| ENEM Challenge (No Images) | 24.42 |
| BLUEX (No Images) | 24.34 |
| OAB Exams | 27.11 |
| Assin2 RTE | 70.86 |
| Assin2 STS | 1.51 |
| FaQuAD NLI | 43.97 |
| HateBR Binary | 40.05 |
| PT Hate Speech Binary | 51.83 |
| tweetSentBR | 30.42 |