Instructions to use colinglab/CLASS_IT-140M with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use colinglab/CLASS_IT-140M with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="colinglab/CLASS_IT-140M")# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("colinglab/CLASS_IT-140M") model = AutoModelForCausalLM.from_pretrained("colinglab/CLASS_IT-140M") - Notebooks
- Google Colab
- Kaggle
- Local Apps
- vLLM
How to use colinglab/CLASS_IT-140M with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "colinglab/CLASS_IT-140M" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "colinglab/CLASS_IT-140M", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/colinglab/CLASS_IT-140M
- SGLang
How to use colinglab/CLASS_IT-140M with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "colinglab/CLASS_IT-140M" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "colinglab/CLASS_IT-140M", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "colinglab/CLASS_IT-140M" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "colinglab/CLASS_IT-140M", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use colinglab/CLASS_IT-140M with Docker Model Runner:
docker model run hf.co/colinglab/CLASS_IT-140M
Model Card for Model ID
Model Description
CLASS-IT is a 140M parameter language model based on the LLaMA architecture.
The model is first pre-trained for 8 epochs on a cleaned version of the BabyLM Challenge strict track dataset. After pre-training, the model is instruction-tuned on two additional datasets (8.7M words total) for 10 epochs :
a conversational dataset derived from Switchboard, and
an educational dataset based on an augmented version of Simple English Wikipedia (to be released soon).
Evaluation
The model has been submitted to the 2025 BabyLM Challenge – Interaction Track: https://huggingface.co/spaces/BabyLM-community/babylm-leaderboard-2025-all-tasks
Citation
This model was introduced in the paper:
“CLASS-IT: Conversational and Lecture-Aligned Small-Scale Instruction Tuning for BabyLMs”
(Capone, Bondielli & Lenci, BabyLM Challange 2025)
📄 ArXiv: 2510.25364
Cite as (BibTeX):
@inproceedings{capone-etal-2025-class,
title = "{CLASS}-{IT}: Conversational and Lecture-Aligned Small-Scale Instruction Tuning for {B}aby{LM}s",
author = "Capone, Luca and
Bondielli, Alessandro and
Lenci, Alessandro",
editor = "Charpentier, Lucas and
Choshen, Leshem and
Cotterell, Ryan and
Gul, Mustafa Omer and
Hu, Michael Y. and
Liu, Jing and
Jumelet, Jaap and
Linzen, Tal and
Mueller, Aaron and
Ross, Candace and
Shah, Raj Sanjay and
Warstadt, Alex and
Wilcox, Ethan Gotlieb and
Williams, Adina",
booktitle = "Proceedings of the First BabyLM Workshop",
month = nov,
year = "2025",
address = "Suzhou, China",
publisher = "Association for Computational Linguistics",
url = "https://aclanthology.org/2025.babylm-main.30/",
pages = "436--444",
ISBN = "TODO"
}
- Downloads last month
- 5,181