Swallow MS 7b Instruct V0.1

🚀 Swallow-MS-7b-v0.1

Our Swallow-MS-7b-v0.1 model is continuously pre-trained based on Mistral-7B-v0.1, with Japanese language data added. It can effectively handle text generation tasks in both English and Japanese, providing users with more language options.

🚀 Quick Start

First, install the additional dependencies in requirements.txt:

pip install -r requirements.txt

Use the instruct model Ver0.1

import torch
from transformers import AutoTokenizer, AutoModelForCausalLM

model_name = "tokyotech-llm/Swallow-MS-7b-instruct-v0.1"
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
tokenizer = AutoTokenizer.from_pretrained(model_name)

device = "cuda"

messages = [
    {"role": "system", "content": "あなたは誠実で優秀な日本人のアシスタントです。"},
    {"role": "user", "content": "東京工業大学の主なキャンパスについて教えてください"}
]

encodeds = tokenizer.apply_chat_template(messages, return_tensors="pt")

model_inputs = encodeds.to(device)
model.to(device)

generated_ids = model.generate(model_inputs, max_new_tokens=128, do_sample=True)
decoded = tokenizer.batch_decode(generated_ids)
print(decoded[0])

✨ Features

• Multilingual Support: Supports both English and Japanese, broadening its application scope.
• Efficient Tokenizer: Employs a tokenizer with an expanded vocabulary based on Japanese data, enabling more efficient text representation and faster inference.

📦 Installation

First install additional dependencies in requirements.txt:

pip install -r requirements.txt

💻 Usage Examples

Basic Usage

import torch
from transformers import AutoTokenizer, AutoModelForCausalLM

model_name = "tokyotech-llm/Swallow-MS-7b-instruct-v0.1"
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
tokenizer = AutoTokenizer.from_pretrained(model_name)

device = "cuda"

messages = [
    {"role": "system", "content": "あなたは誠実で優秀な日本人のアシスタントです。"},
    {"role": "user", "content": "東京工業大学の主なキャンパスについて教えてください"}
]

encodeds = tokenizer.apply_chat_template(messages, return_tensors="pt")

model_inputs = encodeds.to(device)
model.to(device)

generated_ids = model.generate(model_inputs, max_new_tokens=128, do_sample=True)
decoded = tokenizer.batch_decode(generated_ids)
print(decoded[0])

Advanced Usage

When constructing a prompt for the Instruct model, you need to strictly follow the specified template:

<s>[INST] <<SYS>>\n{SYSTEM_PROMPT}\n<</SYS>>\n\n{USER_MESSAGE_1} [/INST] {BOT_MESSAGE_1}</s>[INST] {USER_MESSAGE_2} [/INST] 

Please note that <s> and </s> are special tokens used for the beginning of string (BOS) and end of string (EOS), respectively, while [INST] and [/INST] are considered regular strings.

📚 Documentation

Model Release Updates

We are excited to share the release schedule for our latest models:

• April 26, 2024: Released the Swallow-MS-7b-instruct-v0.1
• March 11, 2024: Released the Swallow-MS-7b-v0.1

This repository provides large language models developed by TokyoTech-LLM.

Model Details

Instruct Model Performance

MT-Bench JA

Turn-Wise Performance

We report overall (i.e., average over scores of the first and second turns), first, and second turn scores.

Overall

First Turn

Second Turn

Comparison to the past model

We only provide the overall score in this section.

Evaluation Benchmarks

MT-Bench JA

We used Japanese MT-Bench to assess the instruction-following capabilities of models. We utilized the following settings:

• Implemantation: FastChat [Zheng+, 2023] (commit #e86e70d0)
• Question: Nejumi LLM-Leaderboard NEO, mtbench_ja_question_v3
• Reference Answer: Nejumi LLM-Leaderboard NEO, mtbench_ja_referenceanswer_v1
• Prompt for Judge: Nejumi LLM-Lederboard NEO, mtbench_ja_prompt_v1
• Judge: gpt-4-1106-preview
• Scoring: Absolute scale normalized to a 0 - 1 range, averaged over five runs.

Training Datasets

Instruction Tuning Ver0.1

The following datasets were used for the instruction tuning:

• OpenAssistant Conversations Dataset was used, where human utterances are included but the responses are not used. Instead, the responses were generated using the Mixtral-8x7B-Instruct-v0.1 model.
• OpenAssistant Conversations Dataset 21k Ja
• OpenAssistant Conversations Dataset 21k En
• Databricks Dolly 15k Ja
• Databricks Dolly 15k En

Please note that some of the data had issues with quality or format, so not all of it was used.

Risks and Limitations

The models released here are still in the early stages of our research and development and have not been tuned to ensure outputs align with human intent and safety considerations.

Acknowledgements

We thank Mistral AI for releasing Mistral 7B v0.1 under an open license for others to build on.

Our project is supported by the ABCI Large-scale Language Model Building Support Program of the National Institute of Advanced Industrial Science and Technology.

📄 License

apache-2.0

👥 Authors

Here are the team members:

• From Okazaki Laboratory, the following members:
  • Naoaki Okazaki
  • Sakae Mizuki
  • Hiroki Iida
  • Mengsay Loem
  • Shota Hirai
  • Kakeru Hattori
  • Masanari Ohi
• From YOKOTA Laboratory, the following members:
  • Rio Yokota
  • Kazuki Fujii
  • Taishi Nakamura
  • Takumi Okamoto
  • Ishida Shigeki

📝 How to cite

If you find our work helpful, please feel free to cite us.

@inproceedings{Fujii:COLM2024,
   title={Continual Pre-Training for Cross-Lingual LLM Adaptation:
Enhancing Japanese Language Capabilities},
   author={Kazuki Fujii and Taishi Nakamura and Mengsay Loem and Hiroki
Iida and Masanari Ohi and Kakeru Hattori and Hirai Shota and Sakae
Mizuki and Rio Yokota and Naoaki Okazaki},
   booktitle="Proceedings of the First Conference on Language Modeling",
   series={COLM},
   pages="(to appear)",
   year="2024",
   month=oct,
   address={University of Pennsylvania, USA},
}

@inproceedings{Okazaki:COLM2024,
   title={Building a Large Japanese Web Corpus for Large Language Models},
   author={Naoaki Okazaki and Kakeru Hattori and Hirai Shota and Hiroki
Iida and Masanari Ohi and Kazuki Fujii and Taishi Nakamura and Mengsay
Loem and Rio Yokota and Sakae Mizuki},
   booktitle="Proceedings of the First Conference on Language Modeling",
   series={COLM},
   pages="(to appear)",
   year="2024",
   month=oct,
   address={University of Pennsylvania, USA},
}