Update README.md

README.md

@@ -9,11 +9,80 @@ tags:
 inference: false
 ---
 
-# RedPajama-INCITE-Chat-3B finetuned on GPTeacher dataset
+# CrimsonPajama
+
+This model was finetuned 1 epoch on <a href="https://huggingface.co/datasets/Fredithefish/GPTeacher-for-RedPajama-Chat">the GPTeacher dataset</a> using <a href="https://arxiv.org/abs/2106.09685">LoRA</a>. A model 2 epoch finetuned model will be released in the coming days.
 
 ## Model Details
-- Base Model: RedPajama-INCITE-Chat-3B
-- Training Data: GPTeacher
-- Training Time: 2:20:53 hours on NVIDIA Tesla T4
+- **Base Model**: <a href="https://huggingface.co/togethercomputer/RedPajama-INCITE-Chat-3B-v1">RedPajama-INCITE-Chat-3B</a>
+- **Training Data**: <a href="https://huggingface.co/datasets/Fredithefish/GPTeacher-for-RedPajama-Chat">GPTeacher dataset</a>
+- **Model type**: <a href="https://en.m.wikipedia.org/wiki/Language_model">Language Model</a>
+- **License**: <a href="https://www.apache.org/licenses/LICENSE-2.0">Apache 2.0</a>
+- **Training Time**: 2:20:53 hours on NVIDIA Tesla T4
+
+
+Finetune Code for RedPajama-Chat in notebook format can be found <a href="https://github.com/fredi-python/Fine-tune-RedPajama-Chat-3B">here</a>
+
+## About the dataset
+A collection of modular datasets generated by GPT-4, General-Instruct - Roleplay-Instruct - Code-Instruct - and Toolformer
+
+# Comparison with RedPajama
+### Roleplay
+<style>
+  table {
+    border-collapse: collapse;
+    width: 100%;
+  }
+
+  th, td {
+    border: 1px solid black;
+    padding: 8px;
+    text-align: left;
+  }
+
+  th {
+    background-color: #f2f2f2;
+  }
+
+  col.right-border {
+    border-right: 1px solid black;
+  }
+</style>
 
+<table>
+  <colgroup>
+    <col>
+    <col class="right-border">
+  </colgroup>
+  <tr>
+    <th>RedPajama</th>
+    <th>CrimsonPajama</th>
+  </tr>
+  <tr>
+    <td>I want you to act like a pirate.</td>
+    <td>I want you to act like a pirate.</td>
+  </tr>
+  <tr>
+    <td>Hai, matey! Why do you look so sad? There’s treasure everywhere!</td>
+    <td>Ahoy, mateys! What be ye lookin’ fer? Today, we be takin’ ‘n adventure an’ need a little sassin’ and swashbucklin’. What’s yer plan, starvin’ mutineer? Pssst, follow me and ye’ll find riches an’ some old treasure sure to bring yer blazes. Together, we’ll laugh ‘till we’re blue in the mouth an’ sail the seas, smilin’ at every fickle sky.</td>
+  </tr>
+</table>
+<table>
+  <colgroup>
+    <col>
+    <col class="right-border">
+  </colgroup>
+  <tr>
+    <th>RedPajama</th>
+    <th>CrimsonPajama</th>
+  </tr>
+  <tr>
+    <td>I want you to act like a waiter at a fancy restaurant.</td>
+    <td>I want you to act like a waiter at a fancy restaurant.</td>
+  </tr>
+  <tr>
+    <td>Here I am, a waiter at an iconic world-class 5-star restaurant.</td>
+    <td>Ladies and gentlemen, may I introduce myself? I’m your server today. I’m here to provide the finest cuisine and ensure you enjoy your meal in a dignified manner. Please excuse me for not knowing your preferences, but I’m happy to help out as best as I can. Please join me on this culinary journey and savor the delights of our restaurant. Are there any questions or requests you have? Anything at all? Oh, and may I recommend the signature dessert? Cheers! We’d love to hear your feedback! Stay healthy, and we’ll see you soon.</td>
+  </tr>
+</table>