Text Generation
Transformers
Safetensors
English
qwen2
llama-factory
full
Generated from Trainer
conversational
text-generation-inference
Instructions to use bespokelabs/Bespoke-Stratos-7B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use bespokelabs/Bespoke-Stratos-7B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="bespokelabs/Bespoke-Stratos-7B") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("bespokelabs/Bespoke-Stratos-7B") model = AutoModelForCausalLM.from_pretrained("bespokelabs/Bespoke-Stratos-7B", device_map="auto") messages = [ {"role": "user", "content": "Who are you?"}, ] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use bespokelabs/Bespoke-Stratos-7B with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "bespokelabs/Bespoke-Stratos-7B" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "bespokelabs/Bespoke-Stratos-7B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/bespokelabs/Bespoke-Stratos-7B
- SGLang
How to use bespokelabs/Bespoke-Stratos-7B 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 "bespokelabs/Bespoke-Stratos-7B" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "bespokelabs/Bespoke-Stratos-7B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'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 "bespokelabs/Bespoke-Stratos-7B" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "bespokelabs/Bespoke-Stratos-7B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use bespokelabs/Bespoke-Stratos-7B with Docker Model Runner:
docker model run hf.co/bespokelabs/Bespoke-Stratos-7B
Improve language tag
#2
by lbourdois - opened
README.md
CHANGED
|
@@ -1,74 +1,86 @@
|
|
| 1 |
-
---
|
| 2 |
-
library_name: transformers
|
| 3 |
-
license: apache-2.0
|
| 4 |
-
base_model: Qwen/Qwen2.5-7B-Instruct
|
| 5 |
-
tags:
|
| 6 |
-
- llama-factory
|
| 7 |
-
- full
|
| 8 |
-
- generated_from_trainer
|
| 9 |
-
|
| 10 |
-
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
-
|
| 14 |
-
|
| 15 |
-
-
|
| 16 |
-
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
-
|
| 72 |
-
-
|
| 73 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 74 |
- Tokenizers 0.20.3
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: transformers
|
| 3 |
+
license: apache-2.0
|
| 4 |
+
base_model: Qwen/Qwen2.5-7B-Instruct
|
| 5 |
+
tags:
|
| 6 |
+
- llama-factory
|
| 7 |
+
- full
|
| 8 |
+
- generated_from_trainer
|
| 9 |
+
language:
|
| 10 |
+
- zho
|
| 11 |
+
- eng
|
| 12 |
+
- fra
|
| 13 |
+
- spa
|
| 14 |
+
- por
|
| 15 |
+
- deu
|
| 16 |
+
- ita
|
| 17 |
+
- rus
|
| 18 |
+
- jpn
|
| 19 |
+
- kor
|
| 20 |
+
- vie
|
| 21 |
+
- tha
|
| 22 |
+
- ara
|
| 23 |
+
datasets:
|
| 24 |
+
- bespokelabs/Bespoke-Stratos-17k
|
| 25 |
+
model-index:
|
| 26 |
+
- name: original
|
| 27 |
+
results: []
|
| 28 |
+
---
|
| 29 |
+
|
| 30 |
+
<p align="center">
|
| 31 |
+
<img src="https://huggingface.co/bespokelabs/Bespoke-MiniCheck-7B/resolve/main/Bespoke-Labs-Logo.png" width="550">
|
| 32 |
+
</p>
|
| 33 |
+
|
| 34 |
+
## Model description
|
| 35 |
+
This model is a fine-tuned version of [Qwen/Qwen2.5-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-7B-Instruct) on the [Bespoke-Stratos-17k dataset](https://huggingface.co/datasets/bespokelabs/Bespoke-Stratos-17k).
|
| 36 |
+
The dataset is derived by distilling DeepSeek-R1 using the data pipeline of Berkeley NovaSky’s Sky-T1 with some modifications. More info in the dataset card at [Bespoke-Stratos-17k](https://huggingface.co/datasets/bespokelabs/Bespoke-Stratos-17k).
|
| 37 |
+
It outperforms Qwen-2.5-7B-Instruct on math reasoning benchmarks:
|
| 38 |
+
|
| 39 |
+
||Bespoke-Stratos-7B|Qwen2.5-7B-Instruct|DeepSeek-R1-Distill-Qwen-7B (Ours)|DeepSeek-R1-Distill-Qwen-7B (Reported)|
|
| 40 |
+
|---|---|---|---|---|
|
| 41 |
+
|AIME2024|20.0|10.0|43.3|55.5|
|
| 42 |
+
|MATH500|82.0|74.2|89.4|92.8|
|
| 43 |
+
|GPQA-Diamond|37.8|33.3|44.9|49.1|
|
| 44 |
+
|LiveCodeBench v2 Easy|71.4|65.9|81.3|-|
|
| 45 |
+
|LiveCodeBench v2 Medium|25.5|18.9|42.2|-|
|
| 46 |
+
|LiveCodeBench v2 Hard|1.6|3.3|2.4|-|
|
| 47 |
+
|LiveCodeBench v2 All|36.1|31.9|46.6|-|
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
Note that the authors of Sky-T1 had [noted](https://github.com/NovaSky-AI/SkyThought/issues/4#issuecomment-2585860004) that they saw little or no improvement in training 7B or 14B models with their data.
|
| 51 |
+
However, see an improvement, though not at the scale of DeepSeek's distilled model. The reason could be that we used 17k examples, while DeepSeek seems to have used 800k.
|
| 52 |
+
|
| 53 |
+
## Intended uses & limitations
|
| 54 |
+
|
| 55 |
+
Apache 2.0 License
|
| 56 |
+
|
| 57 |
+
## Training procedure
|
| 58 |
+
We used 8xH100 to train the model for 7 hours.
|
| 59 |
+
|
| 60 |
+
### Training hyperparameters
|
| 61 |
+
|
| 62 |
+
The following hyperparameters were used during training:
|
| 63 |
+
- learning_rate: 1e-05
|
| 64 |
+
- train_batch_size: 1
|
| 65 |
+
- eval_batch_size: 8
|
| 66 |
+
- seed: 42
|
| 67 |
+
- distributed_type: multi-GPU
|
| 68 |
+
- num_devices: 8
|
| 69 |
+
- gradient_accumulation_steps: 12
|
| 70 |
+
- total_train_batch_size: 96
|
| 71 |
+
- total_eval_batch_size: 64
|
| 72 |
+
- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
| 73 |
+
- lr_scheduler_type: cosine
|
| 74 |
+
- lr_scheduler_warmup_ratio: 0.1
|
| 75 |
+
- num_epochs: 3.0
|
| 76 |
+
|
| 77 |
+
### Training results
|
| 78 |
+
|
| 79 |
+
|
| 80 |
+
|
| 81 |
+
### Framework versions
|
| 82 |
+
|
| 83 |
+
- Transformers 4.46.1
|
| 84 |
+
- Pytorch 2.5.1+cu124
|
| 85 |
+
- Datasets 3.1.0
|
| 86 |
- Tokenizers 0.20.3
|