Update README.md

README.md

@@ -90,4 +90,30 @@ You are focused on providing systematic, well-reasoned responses. Response Struc
 
 if __name__ == "__main__":
     main()
+```
+
+Or alternatively:
+```python
+import torch
+from transformers import pipeline
+
+model_id = "CreitinGameplays/Llama-3.1-8B-R1-experimental"
+
+pipe = pipeline(
+    "text-generation",
+    model=model_id,
+    torch_dtype=torch.bfloat16,
+    device_map="auto"
+)
+
+messages = [{"role": "user", "content": "hello!"}]
+
+outputs = pipe(
+    messages,
+    temperature=0.6,
+    repetition_penalty=1.08,
+    max_new_tokens=2048
+)
+
+print(outputs[0]["generated_text"][-1])
 ```