🚀 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