Views
No views yet

1!pip install -q -U transformers
2!pip install -q -U accelerate
3!pip install -q -U bitsandbytes
4
5from transformers import BitsAndBytesConfig
6import torch
7nb_4bit_config = BitsAndBytesConfig(
8 load_in_4bit=True,
9 bnb_4bit_quant_type="nf4",
10 bnb_4bit_compute_dtype=torch.bfloat16,
11 bnb_4bit_use_double_quant=True
12)
13
14from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
15model = AutoModelForCausalLM.from_pretrained("rhaymison/Llama-portuguese-13b-Luana-v0.2", quantization_config=bnb_config, device_map= {"": 0})
16tokenizer = AutoTokenizer.from_pretrained("rhaymison/Llama-portuguese-13b-Luana-v0.2")
17model.eval()
181
2inputs = tokenizer([f"""<s>[INST] Abaixo está uma instrução que descreve uma tarefa, juntamente com uma entrada que fornece mais contexto.
3Escreva uma resposta que complete adequadamente o pedido.
4### instrução: aja como um professor de matemática e me explique porque 2 + 2 = 4.
5[/INST]"""], return_tensors="pt")
6
7streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
8_ = model.generate(**inputs, streamer=streamer, max_new_tokens=200)
9| Metric | Value |
|---|---|
| Average | 48.83 |
| ENEM Challenge (No Images) | 36.95 |
| BLUEX (No Images) | 32.68 |
| OAB Exams | 33.30 |
| Assin2 RTE | 65.83 |
| Assin2 STS | 42.81 |
| FaQuAD NLI | 40.44 |
| HateBR Binary | 83.62 |
| PT Hate Speech Binary | 54.62 |
| tweetSentBR | 49.25 |