license: apache-2.0
library_name: transformers
pipeline_tag: text-generation
base_model: BSC-LT/salamandra-2b
language:
- bg
- ca
- code
- cs
- cy
- da
- de
- el
- en
- es
- et
- eu
- fi
- fr
- ga
- gl
- hr
- hu
- it
- lt
- lv
- mt
- nl
- nn
- \no
- oc
- pl
- pt
- ro
- ru
- sh
- sk
- sl
- sr
- sv
- uk
Salamandra-2b-fp8 Model Card
This model is the fp8-quantized version of Salamandra-2b.
The model weights are quantized from FP16 to FP8 (8-bit weights) using the FP8 quantization algorithm from NeuralMagic. Inferencing with this model can be done using VLLM.
Salamandra is a highly multilingual model pre-trained from scratch that comes in three different sizes — 2B, 7B and 40B parameters — with their respective base and instruction-tuned variants, promoted and financed by the Government of Catalonia through the Aina Project and the Ministerio para la Transformación Digital y de la Función Pública - Funded by EU – NextGenerationEU within the framework of ILENIA Project with reference 2022/TL22/00215337.
This model card corresponds to the fp8-quantized version of Salamandra-2b.
The entire Salamandra family is released under a permissive Apache 2.0 license.
How to Use
The following example code works under Python 3.9.16, vllm==0.6.3.post1, torch==2.4.0 and torchvision==0.19.0, though it should run on
any current version of the libraries. This is an example of how to create a text completion using the model:
from vllm import LLM, SamplingParams
model_name = "BSC-LT/salamandra-2b-base-fp8"
llm = LLM(model=model_name)
outputs = llm.generate("El mercat del barri ",
sampling_params=SamplingParams(
temperature=0.5,
max_tokens=200)
)
print(outputs[0].outputs[0].text)
Author
International Business Machines (IBM).
Copyright
International Business Machines (IBM).
Contact
For further information, please send an email to langtech@bsc.es.
Acknowledgements
We appreciate the collaboration with IBM in this work. Specifically, the IBM team created fp8-quantized version of the Salamandra-2b model released here.
Disclaimer
Be aware that the model may contain biases or other unintended distortions. When third parties deploy systems or provide services based on this model, or use the model themselves, they bear the responsibility for mitigating any associated risks and ensuring compliance with applicable regulations, including those governing the use of Artificial Intelligence.
Barcelona Supercomputing Center and International Business Machines shall not be held liable for any outcomes resulting from third-party use.