rhaymison's picture
Update README.md
aa0a62d verified
|
Raw
History Blame
3.96 kB
metadata
library_name: transformers
tags:
  - Misral
  - Portuguese
  - 7b
license: apache-2.0
base_model: mistralai/Mistral-7B-Instruct-v0.2
datasets:
  - pablo-moreira/gpt4all-j-prompt-generations-pt
  - rhaymison/superset
language:
  - pt
pipeline_tag: text-generation

Mistral-portuguese-luana-7b

Bode Logo

This model was trained with a superset of 200,000 instructions in Portuguese. The model comes to help fill the gap in models in Portuguese. Tuned from the Mistral 7b in Portuguese, the model was adjusted mainly for instructional tasks.

How to use

You can use the model in its normal form up to 4-bit quantization. Below we will use both approaches. Remember that verbs are important in your prompt. Tell your model how to act or behave so that you can guide them along the path of their response. Important points like these help models (even smaller models like 7b) to perform much better.

!pip install -q -U transformers
!pip install -q -U accelerate
!pip install -q -U bitsandbytes

from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
model = AutoModelForCausalLM.from_pretrained("rhaymison/Mistral-portuguese-luana-7b", device_map= {"": 0})
tokenizer = AutoTokenizer.from_pretrained("rhaymison/Mistral-portuguese-luana-7b")
model.eval()

You can use with Pipeline but in this example i will use such as Streaming


inputs = tokenizer([f"""<s>[INST] 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.
### instrução: aja como um professor de matemática e me explique porque 2 + 2 = 4.
[/INST]"""], return_tensors="pt")

streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
_ = model.generate(**inputs, streamer=streamer, max_new_tokens=200)

If you are having a memory problem such as "CUDA Out of memory", you should use 4-bit or 8-bit quantization. For the complete model in colab you will need the A100. If you want to use 4bits or 8bits, T4 or L4 will already solve the problem.

4bits example

from transformers import BitsAndBytesConfig
import torch
nb_4bit_config = BitsAndBytesConfig(
    load_in_4bit=True,
    bnb_4bit_quant_type="nf4",
    bnb_4bit_compute_dtype=torch.bfloat16,
    bnb_4bit_use_double_quant=True
)

model = AutoModelForCausalLM.from_pretrained(
    base_model,
    quantization_config=bnb_config,
    device_map={"": 0}
)

Benchmark test

benchmark: portuguese-benchmark-datasets/BLUEX

Metric Valor
Hits 485
Erros 600
Total Questions 1.085
Percentage 44.70%

benchmark: benchmark-summarization

Metric Valor
ROUGE-1 0.62
ROUGE-2 0.49
ROUGE-L 0.58

benchmark: portuguese-benchmark-datasets/HateBR_offensive_binary

Metric Valor
Hits 795
Erros 325
Total Questions 1120
Percentage 70.98%

Comments

Any idea, help or report will always be welcome.

email: rhaymisoncristian@gmail.com