Croissantllmbase GGUF

🚀 CroissantLLM - Base GGUF (190k steps, Final version)

CroissantLLM is a high - performance, fully open - sourced bilingual model. This specific version corresponds to the checkpoint after 190k steps (2.99 T) tokens, aiming to offer excellent performance on consumer - grade local hardware.

To play with the final model, we recommend using the Chat version: https://huggingface.co/croissantllm/CroissantLLMChat-v0.1.

https://arxiv.org/abs/2402.00786

🚀 Quick Start

This model is part of the CroissantLLM initiative, and corresponds to the checkpoint after 190k steps (2.99 T) tokens. You can access the chat version via the link provided above to interact with the final model.

✨ Features

• Bilingual Capability: Pretrained on a set of 3T English and French tokens, offering high - performance bilingual processing.
• Innovative Training Approach: Pioneers the approach of training an intrinsically bilingual model with a 1:1 English - to - French pretraining data ratio, a custom tokenizer, and bilingual finetuning datasets.
• Transparency: Releases training datasets, codebases, dozens of checkpoints, fine - tuned Chat models, and strong translation models.

📦 Installation

No specific installation steps are provided in the original document.

💻 Usage Examples

Basic Usage

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "croissantllm/CroissantLLMBase"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")

inputs = tokenizer("I am so tired I could sleep right now. -> Je suis si fatigué que je pourrais m'endormir maintenant.\nHe is heading to the market. -> Il va au marché.\nWe are running on the beach. ->", return_tensors="pt").to(model.device)
tokens = model.generate(**inputs, max_length=100, do_sample=True, top_p=0.95, top_k=60, temperature=0.3)
print(tokenizer.decode(tokens[0]))

# remove bos token
inputs = tokenizer("Capitales: France -> Paris, Italie -> Rome, Allemagne -> Berlin, Espagne ->", return_tensors="pt", add_special_tokens=True).to(model.device)
tokens = model.generate(**inputs, max_length=100, do_sample=True, top_p=0.95, top_k=60)
print(tokenizer.decode(tokens[0]))

📚 Documentation

Abstract

We introduce CroissantLLM, a 1.3B language model pretrained on a set of 3T English and French tokens, to bring to the research and industrial community a high - performance, fully open - sourced bilingual model that runs swiftly on consumer - grade local hardware. To that end, we pioneer the approach of training an intrinsically bilingual model with a 1:1 English - to - French pretraining data ratio, a custom tokenizer, and bilingual finetuning datasets. We release the training dataset, notably containing a French split with manually curated, high - quality, and varied data sources. To assess performance outside of English, we craft a novel benchmark, FrenchBench, consisting of an array of classification and generation tasks, covering various orthogonal aspects of model performance in the French Language. Additionally, rooted in transparency and to foster further Large Language Model research, we release codebases, and dozens of checkpoints across various model sizes, training data distributions, and training steps, as well as fine - tuned Chat models, and strong translation models. We evaluate our model through the FMTI framework, and validate 81% of the transparency criteria, far beyond the scores of even most open initiatives. This work enriches the NLP landscape, breaking away from previous English - centric work in order to strengthen our understanding of multilinguality in language models.

Citation

Our work can be cited as:

@misc{faysse2024croissantllm,
      title={CroissantLLM: A Truly Bilingual French - English Language Model}, 
      author={Manuel Faysse and Patrick Fernandes and Nuno M. Guerreiro and António Loison and Duarte M. Alves and Caio Corro and Nicolas Boizard and João Alves and Ricardo Rei and Pedro H. Martins and Antoni Bigata Casademunt and François Yvon and André F. T. Martins and Gautier Viaud and Céline Hudelot and Pierre Colombo},
      year={2024},
      eprint={2402.00786},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}

Usage

This model is a base model, that is, it is not finetuned for Chat function and works best with few - shot prompting strategies.

📄 License

This model is released under the MIT license.