Mistral Nemo Japanese Instruct 2408

🚀 Mistral-Nemo-Japanese-Instruct-2408

這是一個基於mistralai/Mistral-Nemo-Instruct-2407的日語持續預訓練模型，可用於文本生成相關任務。

🚀 快速開始

在使用該模型前，請確保通過以下命令更新你的transformers庫：

pip install --upgrade transformers

💻 使用示例

基礎用法

from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer

model = AutoModelForCausalLM.from_pretrained("cyberagent/Mistral-Nemo-Japanese-Instruct-2408", device_map="auto", torch_dtype="auto")
tokenizer = AutoTokenizer.from_pretrained("cyberagent/Mistral-Nemo-Japanese-Instruct-2408")
streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)

messages = [
    {"role": "system", "content": "あなたは親切なAIアシスタントです。"},
    {"role": "user", "content": "AIによって私たちの暮らしはどのように変わりますか？"}
]

input_ids = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)
output_ids = model.generate(input_ids,
                            max_new_tokens=1024,
                            temperature=0.5,
                            streamer=streamer)

📚 詳細文檔

提示格式

本模型採用ChatML格式，示例如下：

<s><|im_start|>system
あなたは親切なAIアシスタントです。<|im_end|>
<|im_start|>user
AIによって私たちの暮らしはどのように変わりますか？<|im_end|>
<|im_start|>assistant

📄 許可證

本模型使用的許可證為Apache-2.0。

👨‍💻 作者

Ryosuke Ishigami

📖 引用方式

如果你在研究中使用了本模型，請使用以下BibTeX引用：

@misc{cyberagent-mistral-nemo-japanese-instruct-2408,
      title={Mistral-Nemo-Japanese-Instruct-2408},
      url={https://huggingface.co/cyberagent/Mistral-Nemo-Japanese-Instruct-2408},
      author={Ryosuke Ishigami},
      year={2024},
}