jisx's picture
Upload README.md with huggingface_hub
0a58eb2 verified
|
raw
history blame
1.28 kB
metadata
library_name: transformers
pipeline_tag: text-generation
language:
  - multilingual
tags:
  - generation
  - question answering
  - instruction tuning
datasets:
  - MBZUAI/Bactrian-X
license: cc-by-nc-4.0

Model Description

This HF repository hosts instruction fine-tuned multilingual BLOOM model using the parallel instruction dataset called Bactrain-X in 52 languages. We progressively add a language during instruction fine-tuning at each time, and train 52 models in total. Then, we evaluate those models in three multilingual benchmarks.

Please refer to our paper for more details.

Instruction tuning details

  • Base model: BLOOM 7B1
  • Instruction languages: English, Chinese, Afrikaans, Arabic, Azerbaijani, Bengali, Czech, German, Spanish, Estonian
  • Instruction language codes: en, zh, af, ar, az, bn, cs, de, es, et
  • Training method: full-parameter fine-tuning.

Usage

The model checkpoint should be loaded using transformers library.

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("MaLA-LM/lucky52-bloom-7b1-no-10")
model = AutoModelForCausalLM.from_pretrained("MaLA-LM/lucky52-bloom-7b1-no-10")

Citation

@article{
}