license: apache-2.0
language:
- en
- es
- de
- fr
- it
pipeline_tag: text-generation
Occiglot-7B-EU5
Occiglot-7B-EU5-Instruct is a the instruct version of occiglot-7b-eu5, a generative language model with 7B parameters supporting the top-5 EU languages (English, Spanish, French, German, and Italian) and trained by the Occiglot Research Collective. It is based on Mistral-7B-v0.1 and trained on 400M tokens of additional multilingual and code instructions with a block size of 8,192 tokens per sample. Note that the model was not safety aligned and might generate problematic outputs.
This is the first release of an ongoing open research project for multilingual language models. If you want to train a model for your own language or are working on evaluations, please contact us or join our Discord server. We are open for collaborations!
Model details
- Continued-pretraining from: Mistral-7B-v0.1
- Model type: Causal decoder-only transformer language model
- Languages: English, Spanish, French, German, Italian, and code.
- License: Apache 2.0
- Compute resources: DFKI cluster
- Contributors: Manuel Brack, Patrick Schramowski, Pedro Ortiz, Malte Ostendorff, Fabio Barth, Georg Rehm, Kristian Kersting
- Research labs: Occiglot with support from SAINT and SLT
- Contact: Discord hello@occiglot.org
How to use
The model was trained using the chatml instruction template. You can use the transformers chat template feature for interaction. Since the generation relies on some randomness, we set a seed for reproducibility:
>>> from transformers import AutoTokenizer, MistralForCausalLM, set_seed
>>> tokenizer = AutoTokenizer.from_pretrained("occiglot/occiglot-7b-de-en-instruct")
>>> model = MistralForCausalLM.from_pretrained('occiglot/occiglot-7b-de-en-instruct') # You may want to use bfloat16 and/or move to GPU here
>>> set_seed(42)
>>> messages = [
>>> {"role": "system", 'content': 'You are a helpful assistant. Please give short and concise answers.'},
>>> {"role": "user", "content": "Wer ist der deutsche Bundeskanzler?"},
>>> ]
>>> tokenized_chat = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_dict=False, return_tensors='pt',)
>>> set_seed(42)
>>> outputs = model.generate(tokenized_chat.to('cuda'), max_new_tokens=200,)
>>> tokenizer.decode(out[0][len(tokenized_chat[0]):])
'Der deutsche Bundeskanzler ist Olaf Scholz.'
Dataset
The training data was split evenly amongst the 5 languages based on the total number of tokens. We would like to thank Disco Research and Björn Plüster for making their dataset available to us.
English and Code
German
- DiscoLM German Dataset
- OASST-2 (German subset)
- Aya-Dataset (German subset)
Spanish
- Mentor-ES
- Squad-es
- OASST-2 (Spanish subset)
- Aya-Dataset (Spanish subset)
French
- Bactrian-X (French subset)
- AI-Society Translated (French subset)
- GT-Dorimiti
- OASST-2 (French subset)
- Aya-Dataset (French subset)
Italian
- Quora-IT-Baize
- Stackoverflow-IT-Vaize
- Camoscio
- OASST-2 (Italian subset)
- Aya-Dataset (Italian subset)
Training settings
- Full instruction fine-tuning on 8xH100.
- 0.6 - 4 training epochs (depending on dataset sampling).
- Framework: axolotl
- Precision: bf16
- Optimizer: AdamW
- Global batch size: 128 (with 8192 context length)
- Cosine Annealing with Warmup
Tokenizer
Tokenizer is unchanged from Mistral-7B-v0.1.
Evaluation
Preliminary evaluation results can be found below. Please note that the non-English results are based on partially machine-translated datasets and English prompts (Belebele and Okapi framework) and thus should be interpreted with caution, e.g., biased towards English model performance. Currently, we are working on more suitable benchmarks for Spanish, French, German, and Italian.
Evaluation results
Acknowledgements
The pre-trained model training was supported by a compute grant at the 42 supercomputer which is a central component in the development of hessian AI, the AI Innovation Lab (funded by the Hessian Ministry of Higher Education, Research and the Art (HMWK) & the Hessian Ministry of the Interior, for Security and Homeland Security (HMinD)) and the AI Service Centers (funded by the German Federal Ministry for Economic Affairs and Climate Action (BMWK)). The curation of the training data is partially funded by the German Federal Ministry for Economic Affairs and Climate Action (BMWK) through the project OpenGPT-X (project no. 68GX21007D).