Update README.md

README.md

@@ -3,8 +3,31 @@ license: apache-2.0
 ---
 
 Obtained from freecs/ThetaWave-7B after SFT fine tuning.
----
+
 Open-Orca/SlimOrca datasets were used.
----
+
 If the performance is better, more model details will be released...
----
+
+
+Try it:
+```
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+device = "cuda" # the device to load the model onto
+
+model = AutoModelForCausalLM.from_pretrained("Liangmingxin/ThetaWave-7B-sft")
+tokenizer = AutoTokenizer.from_pretrained("Liangmingxin/ThetaWave-7B-sft")
+
+messages = [
+    {"role": "user", "content": "Who are you?"},
+]
+
+encodeds = tokenizer.apply_chat_template(messages, return_tensors="pt")
+
+model_inputs = encodeds.to(device)
+model.to(device)
+
+generated_ids = model.generate(model_inputs, max_new_tokens=1000, do_sample=True)
+decoded = tokenizer.batch_decode(generated_ids)
+print(decoded[0])
+```