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},
}