Update README.md

README.md

@@ -9,14 +9,35 @@ tags:
 - unsloth
 - mistral
 - trl
+- sft
 ---
+# Jokestral
 
-# Uploaded  model
+This model was created by fine-tuning `unsloth/mistral-7b-v0.3-bnb-4bit` on [Short jokes dataset](https://www.kaggle.com/datasets/abhinavmoudgil95/short-jokes).
+So the only purpose of this model is the generation of cringe jokes. </br>
+Just write the first few words and get your joke.
 
-- **Developed by:** SantaBot
-- **License:** apache-2.0
-- **Finetuned from model :** unsloth/mistral-7b-v0.3-bnb-4bit
+# Usage
+
+[**Goodle Colab example**](https://colab.research.google.com/drive/13N1O-fq-vjr8FUrsUU6f24fPpyf0ZwOS#scrollTo=UBSG1UTV85Vq)
+
+```
+pip install transformers
+pip install --no-deps "trl<0.9.0" peft accelerate bitsandbytes
+```
+```
+from transformers import AutoTokenizer,AutoModelForCausalLM
+model = AutoModelForCausalLM.from_pretrained("SantaBot/Jokestral_16bit",)
+tokenizer = AutoTokenizer.from_pretrained("SantaBot/Jokestral_16bit")
+inputs = tokenizer(
+[
+    "My doctor" # YOUR PROMPT HERE
+], return_tensors = "pt").to("cuda")
+outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)
+tokenizer.batch_decode(outputs)
+```
+
+**The output should be something like** : </br>
+`['<s> My doctor told me I have to stop m4sturb4t1ng. I asked him why and he said ""Because I\'m trying to examine you.""\n</s>']`
 
-This mistral model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
 
-[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)