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 AutoModelForCausalLM, AutoTokenizer, TextStreamer
6model = AutoModelForCausalLM.from_pretrained("rhaymison/Mistral-portuguese-luana-7b", device_map= {"": 0})
7tokenizer = AutoTokenizer.from_pretrained("rhaymison/Mistral-portuguese-luana-7b")
8model.eval()
91
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
7inputs.to(model.device)
8
9streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
10_ = model.generate(**inputs, streamer=streamer, max_new_tokens=200)
111from transformers import BitsAndBytesConfig
2import torch
3nb_4bit_config = BitsAndBytesConfig(
4 load_in_4bit=True,
5 bnb_4bit_quant_type="nf4",
6 bnb_4bit_compute_dtype=torch.bfloat16,
7 bnb_4bit_use_double_quant=True
8)
9
10model = AutoModelForCausalLM.from_pretrained(
11 base_model,
12 quantization_config=bnb_config,
13 device_map={"": 0}
14)
15
| Metric | Value |
|---|---|
| Average | 64.27 |
| ENEM Challenge (No Images) | 58.64 |
| BLUEX (No Images) | 47.98 |
| OAB Exams | 38.82 |
| Assin2 RTE | 90.63 |
| Assin2 STS | 75.81 |
| FaQuAD NLI | 57.79 |
| HateBR Binary | 77.24 |
| PT Hate Speech Binary | 68.50 |
| tweetSentBR | 63 |