Update README.md

README.md

@@ -25,6 +25,44 @@ The following models were included in the merge:
 
 This model was merged using the [DELLA](https://arxiv.org/abs/2406.11617) merge method
 
+Cuda
+
+```
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model_name = "muzerai/SmolLM3-3B-Merged-AIJOAH"
+device = "cuda"  # for GPU usage or "cpu" for CPU usage
+
+# load the tokenizer and the model
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(
+    model_name,
+).to(device)
+
+# prepare the model input
+prompt = "Give me a brief explanation of gravity in simple terms."
+messages_think = [
+    {"role": "user", "content": prompt}
+]
+
+text = tokenizer.apply_chat_template(
+    messages_think,
+    tokenize=False,
+    add_generation_prompt=True,
+)
+model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
+
+# Generate the output
+generated_ids = model.generate(**model_inputs, max_new_tokens=32768)
+
+# Get and decode the output
+output_ids = generated_ids[0][len(model_inputs.input_ids[0]) :]
+print(tokenizer.decode(output_ids, skip_special_tokens=True))
+
+```
+
+
+MAC
 ```
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer