Instructions to use stabilityai/stablelm-2-1_6b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use stabilityai/stablelm-2-1_6b with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="stabilityai/stablelm-2-1_6b")# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-1_6b") model = AutoModelForCausalLM.from_pretrained("stabilityai/stablelm-2-1_6b") - Notebooks
- Google Colab
- Kaggle
- Local Apps
- vLLM
How to use stabilityai/stablelm-2-1_6b with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "stabilityai/stablelm-2-1_6b" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "stabilityai/stablelm-2-1_6b", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/stabilityai/stablelm-2-1_6b
- SGLang
How to use stabilityai/stablelm-2-1_6b 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 "stabilityai/stablelm-2-1_6b" \ --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": "stabilityai/stablelm-2-1_6b", "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 "stabilityai/stablelm-2-1_6b" \ --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": "stabilityai/stablelm-2-1_6b", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use stabilityai/stablelm-2-1_6b with Docker Model Runner:
docker model run hf.co/stabilityai/stablelm-2-1_6b
Training config link is broken
Hi, in the section Training procedure the training config link is missing.
I would like to continue pre-training. Is there any additional advice?
same reason here, the config seams to be missing in the repo :-)
any news?
anyone here?
@jon-tow hello?
I leave it open to fix the Model card
Hi @davidgortega ! Sorry, I was swamped and missed this. Let me know if you have any questions about the config; I can try to answer them ASAP.
Thanks!
Im planing to continue pre-training with a dataset of 100M tokens for two epochs. Do you think it would be enough to learn it?
hello new here kind of wanted to learn more and figure out some things some of this is way over my head I feel like I need a book for dummy's, but in your bust words how do i get the lib for this and use it I get lost alot 🤔
@davidgortega re:
Im planing to continue pre-training with a dataset of 100M tokens for two epochs. Do you think it would be enough to learn it?
If the domain of your data is relatively close to the pre-training dataset (see datasets metadata), it should be enough. Otherwise, it is hard to tell 😅 I'd also suggest fine-tuning the released checkpoint as opposed to continued pre-training from the pre-cooldown version since it's only for 200M tokens.
@jon-tow thanks for the reply.
Its wiki data in a specific domain (like fandom). I hope it works.
The problem of fine-tuning after the cool down is that training raw data with an empty prompt alone does not work as I expect. I have to combine empty prompts with synthetic instruct data generated from my corpus to learn a little and the outcome hallucinates a bit too much. Maybe you have a recipe?