🚀 Mistral-portuguese-luana-7b-chat
本模型基于25万个葡萄牙语聊天数据集进行训练,旨在填补葡萄牙语模型的空白。它在Mistral 7B的基础上进行微调,主要用于聊天场景。
🚀 快速开始
硬件要求
- 完整模型:需要A100 GPU。
- 半精度模型:L4 GPU即可。
- 8位或4位量化:T4或V100 GPU可满足需求。
你可以使用正常形式的模型,也可以进行4位量化。以下将展示这两种使用方式。请记住,在提示词中使用合适的动词很重要,明确告知模型如何行动或表现,能引导其生成更好的回复。这些要点能帮助模型(即使是7B这样的小模型)表现得更出色。
安装依赖
!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-chat", device_map= {"": 0})
tokenizer = AutoTokenizer.from_pretrained("rhaymison/Mistral-portuguese-luana-7b-chat")
model.eval()
流式输出示例
inputs = tokenizer([f"""<s>[INST] Estou pensando em levar minha esposa e meu gatinha para fazer algo diferente. Você teria alguma coisa em mente ?.
[/INST]"""], return_tensors="pt")
inputs.to(model.device)
streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
_ = model.generate(**inputs, streamer=streamer, max_new_tokens=128)
内存问题处理
如果遇到“CUDA Out of memory”这样的内存问题,你可以使用4位或8位量化。在Colab上运行完整模型需要A100 GPU,若使用4位或8位量化,T4或L4 GPU就能解决问题。
4位量化示例
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}
)
💻 使用示例
LangChain示例
用户: Ola Luana, estou precisando de umas dicas para o final de semana.
Poderia me ajudar ?
Luana: Olá! Claro, ficarei feliz em ajudá-lo. O que você tem em mente para o final de semana?
用户: Estou pensando em levar minha esposa e meu gatinha para fazer algo diferente.
Você teria alguma coisa em mente ?
Luana: Claro! Você pode considerar fazer um piquenique na praia, visitar um museu
ou fazer uma caminhada na natureza. O que você e sua esposa gostam de fazer juntos?
用户: Eu e minha esposa gostamos muito de caminhadas.
Temos uma mochila que carregamos nosso gatinha para todo lado.
Luana: Parece ótimo! Você pode considerar fazer uma caminhada em um parque local ou reserva natural. Você
também pode pesquisar trilhas mais longas para fazer em um fim de semana.
Além disso, você pode trazer um piquenique com você para saborear durante a caminhada.
用户: Otimas ideias voce tem me dado. O que será que é gostoso levar em um piquenique ?
Luana: Você pode trazer sanduíches, frutas, vegetais, queijo e biscoitos.
Você também pode trazer uma garrafa de água ou um refrigerante para beber.
Além disso, você pode trazer um saco de lixo para descartar resíduos de maneira responsável.
用户: Olha só que dica bacana. Realmente devo levar uma sacola para nao descartar
nada na Natureza e assim me divertir com minha esposa.
Luana: Sim, é importante descartar resíduos de maneira responsável e proteger o meio ambiente.
Você pode considerar fazer um piquenique em um parque local ou reserva natural onde
possa descartar resíduos de maneira responsável.
用户: Perfeito. Muito Obrigado Luana. Otimas dicas. Até a proxima !
Luana: De nada! Fique em contato se precisar de mais ajuda. Tenha um ótimo fim de semana!
📚 详细文档
详细结果可查看此处
指标 |
数值 |
平均值 |
65.03 |
ENEM Challenge (No Images) |
59.13 |
BLUEX (No Images) |
49.24 |
OAB Exams |
36.58 |
Assin2 RTE |
90.47 |
Assin2 STS |
76.55 |
FaQuAD NLI |
66.75 |
HateBR Binary |
77.46 |
PT Hate Speech Binary |
69.45 |
tweetSentBR |
59.63 |
📄 许可证
本模型采用Apache-2.0许可证。
交流与反馈
欢迎提出任何想法、提供帮助或反馈问题。
- 邮箱:rhaymisoncristian@gmail.com