GLM-5.3 is live. Run Z.ai's latest model on Friendli Model APIs. Try it today ➜
HuggingFaceTB
Available on FriendliAI
Run this model inference on single tenant GPU with unmatched speed and reliability at scale.
Run this model inference with full control and performance in your environment.
Model Details
Model Provider
HuggingFaceTB
Model Tree
Input Modalities
Output Modalities
Supported Functionality
SmolLM is a series of state-of-the-art small language models available in three sizes: 135M, 360M, and 1.7B parameters. These models are built on Cosmo-Corpus, a meticulously curated high-quality training dataset. Cosmo-Corpus includes Cosmopedia v2 (28B tokens of synthetic textbooks and stories generated by Mixtral), Python-Edu (4B tokens of educational Python samples from The Stack), and FineWeb-Edu (220B tokens of deduplicated educational web samples from FineWeb). SmolLM models have shown promising results when compared to other models in their size categories across various benchmarks testing common sense reasoning and world knowledge. For detailed information on training, benchmarks and performance, please refer to our full blog post.
This is the SmolLM-360M
bash
pip install transformers
python
# pip install transformersfrom transformers import AutoModelForCausalLM, AutoTokenizercheckpoint = "HuggingFaceTB/SmolLM-360M"device = "cuda" # for GPU usage or "cpu" for CPU usagetokenizer = AutoTokenizer.from_pretrained(checkpoint)# for multiple GPUs install accelerate and do `model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map="auto")`model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)inputs = tokenizer.encode("def print_hello_world():", return_tensors="pt").to(device)outputs = model.generate(inputs)print(tokenizer.decode(outputs[0]))* _Using `torch.bfloat16`_```python# pip install accelerateimport torchfrom transformers import AutoTokenizer, AutoModelForCausalLMcheckpoint = "HuggingFaceTB/SmolLM-360M"tokenizer = AutoTokenizer.from_pretrained(checkpoint)# for fp16 use `torch_dtype=torch.float16` insteadmodel = AutoModelForCausalLM.from_pretrained(checkpoint, device_map="auto", torch_dtype=torch.bfloat16)inputs = tokenizer.encode("def print_hello_world():", return_tensors="pt").to("cuda")outputs = model.generate(inputs)print(tokenizer.decode(outputs[0]))
bash
>>> print(f"Memory footprint: {model.get_memory_footprint() / 1e6:.2f} MB")Memory footprint: 723.56 MB
bitsandbytespython
# pip install bitsandbytes acceleratefrom transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig# to use 4bit use `load_in_4bit=True` insteadquantization_config = BitsAndBytesConfig(load_in_8bit=True)checkpoint = "HuggingFaceTB/SmolLM-360M"tokenizer = AutoTokenizer.from_pretrained(checkpoint)model = AutoModelForCausalLM.from_pretrained(checkpoint, quantization_config=quantization_config)inputs = tokenizer.encode("def print_hello_world():", return_tensors="pt").to("cuda")outputs = model.generate(inputs)print(tokenizer.decode(outputs[0]))
bash
>>> print(f"Memory footprint: {model.get_memory_footprint() / 1e6:.2f} MB")# load_in_8bitMemory footprint: 409.07 MB# load_in_4bit>>> print(f"Memory footprint: {model.get_memory_footprint() / 1e6:.2f} MB")Memory footprint: 251.79 MB
While SmolLM models have been trained on a diverse dataset including educational content and synthetic texts, they have limitations. The models primarily understand and generate content in English. They can produce text on a variety of topics, but the generated content may not always be factually accurate, logically consistent, or free from biases present in the training data. These models should be used as assistive tools rather than definitive sources of information. Users should always verify important information and critically evaluate any generated content. For a more comprehensive discussion of the models' capabilities and limitations, please refer to our full blog post..
This repository contains a converted version of our latest trained model. We've noticed a small performance difference between this converted checkpoint (transformers) and the original (nanotron). We're currently working to resolve this issue.
bash
@misc{allal2024SmolLM,title={SmolLM - blazingly fast and remarkably powerful},author={Loubna Ben Allal and Anton Lozhkov and Elie Bakouch and Leandro von Werra and Thomas Wolf},year={2024},}