Update README.md

README.md

@@ -32,3 +32,25 @@ Technical details can be found in [Kaggle Discussion](https://www.kaggle.com/com
 # Dataset
 - [Our first stage SFT dataset](https://huggingface.co/datasets/RabotniKuma/Fast-Math-R1-SFT)
 - [Our second stage GRPO dataset](https://huggingface.co/datasets/RabotniKuma/Fast-Math-R1-GRPO)
+
+# Inference
+## vLLM
+```python
+from vllm import LLM, SamplingParams
+
+
+vllm_engine = LLM(
+    model='RabotniKuma/Fast-Math-R1-14B',
+    max_model_len=8192,
+    gpu_memory_utilization=0.9,
+    trust_remote_code=True,
+)
+sampling_params = SamplingParams(
+    temperature=1.0,
+    top_p=0.90,
+    min_p=0.05,
+    max_tokens=8192,
+    stop='</think>',  # Important: early stop at </think> to save output tokens
+)
+vllm_engine.generate('1+1=', sampling_params=sampling_params)
+```