Update README.md

README.md

@@ -27,4 +27,22 @@ tags:
 <prompt> (without the <>)
 
 ### Response:
-```
+```
+
+## Sample Code
+
+```python
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+torch.set_default_device("cuda")
+
+model = AutoModelForCausalLM.from_pretrained("ibivibiv/strix-rufipes-70b", torch_dtype="auto", device_config='auto')
+tokenizer = AutoTokenizer.from_pretrained("ibivibiv/strix-rufipes-70b", trust_remote_code=True)
+
+inputs = tokenizer("Create a plan for developing the game of snake in python using pygame.", return_tensors="pt", return_attention_mask=False)
+
+outputs = model.generate(**inputs, max_length=200)
+text = tokenizer.batch_decode(outputs)[0]
+print(text)
+```