Update README.md

README.md

@@ -1,6 +1,6 @@
 ---
 license: mit
-base_model: BramVanroy/fietje-2b-instruct
+base_model: BramVanroy/fietje-2-instruct
 tags:
 - trl
 - fietje
@@ -10,7 +10,7 @@ datasets:
 - BramVanroy/ultra_feedback_dutch_cleaned
 - BramVanroy/orca_dpo_pairs_dutch_cleaned
 model-index:
-- name: fietje-2b-chat
+- name: fietje-2-chat
   results: []
 pipeline_tag: text-generation
 inference: false
@@ -19,27 +19,27 @@ language:
 ---
 
 <p align="center" style="margin:0;padding:0">
-  <img src="https://huggingface.co/BramVanroy/fietje-2b-chat/resolve/main/img/fietje-2b-banner-rounded.png" alt="Fietje banner" width="800" style="margin-left:'auto' margin-right:'auto' display:'block'"/>
+  <img src="https://huggingface.co/BramVanroy/fietje-2-chat/resolve/main/img/fietje-2b-banner-rounded.png" alt="Fietje banner" width="800" style="margin-left:'auto' margin-right:'auto' display:'block'"/>
 </p>
 
 <div style="margin:auto; text-align:center">
-  <h1 style="margin-bottom: 0">Fietje 2B Chat</h1>
+  <h1 style="margin-bottom: 0">Fietje 2 Chat</h1>
   <em>An open and efficient LLM for Dutch</em>
 </div>
 
 <blockquote class="tip" style="padding: 1.5em; border: 0">
   <p align="center" style="text-align: center; margin: 0">
-    <a href="https://huggingface.co/BramVanroy/fietje-2b">👱‍♀️ Base version</a> -
-    <a href="https://huggingface.co/BramVanroy/fietje-2b-instruct">🤖 Instruct version</a> -
-    <a href="https://huggingface.co/BramVanroy/fietje-2b-chat">💬 Chat version</a> (this one) -
-    <a href="https://huggingface.co/BramVanroy/fietje-2b-chat-GGUF">🚀 GGUF of Chat</a>
+    <a href="https://huggingface.co/BramVanroy/fietje-2">👱‍♀️ Base version</a> -
+    <a href="https://huggingface.co/BramVanroy/fietje-2-instruct">🤖 Instruct version</a> -
+    <a href="https://huggingface.co/BramVanroy/fietje-2-chat">💬 Chat version</a> (this one) -
+    <a href="https://huggingface.co/BramVanroy/fietje-2-chat-GGUF">🚀 GGUF of Chat</a>
   </p>
   <p align="center" style="text-align: center; margin: 0">
-    <a href="https://huggingface.co/spaces/BramVanroy/fietje-2b"><strong>Chat with Fietje here!</strong></a>
+    <a href="https://huggingface.co/spaces/BramVanroy/fietje-2"><strong>Chat with Fietje here!</strong></a>
   </p>
 </blockquote>
 
-This is the chat version of Fietje, a DPO-tuned (aligned) continuation on [the instruct version](https://huggingface.co/BramVanroy/fietje-2b-instruct). Fietje is an adapated version of [microsoft/phi-2](https://huggingface.co/microsoft/phi-2), tailored to Dutch text generation by training on 28B tokens. It is small and efficient with a size of 2.7 billion parameters while performing almost on par with more powerful Dutch LLMs of twice its size like [GEITje 7B Ultra](https://huggingface.co/BramVanroy/GEITje-7B-ultra).
+This is the chat version of Fietje, a DPO-tuned (aligned) continuation on [the instruct version](https://huggingface.co/BramVanroy/fietje-2-instruct). Fietje is an adapated version of [microsoft/phi-2](https://huggingface.co/microsoft/phi-2), tailored to Dutch text generation by training on 28B tokens. It is small and efficient with a size of 2.7 billion parameters while performing almost on par with more powerful Dutch LLMs of twice its size like [GEITje 7B Ultra](https://huggingface.co/BramVanroy/GEITje-7B-ultra).
 
 A thorough description of the creation and evaluation of Fietje as well as usage examples are available in [this Github repository](https://github.com/BramVanroy/fietje).
 
@@ -49,12 +49,12 @@ The same limitations as [phi-2](https://huggingface.co/microsoft/phi-2#limitatio
 
 ## Training and evaluation data
 
-Fietje 2B instruct was finetuned from [the instruct model](https://huggingface.co/BramVanroy/fietje-2b-instruct) on the following datasets. Number of training samples per dataset given in brackets, totalling 18,653 samples.
+Fietje 2 Chat was finetuned from [the instruct model](https://huggingface.co/BramVanroy/fietje-2-instruct) on the following datasets. Number of training samples per dataset given in brackets, totalling 18,653 samples.
 
 - [BramVanroy/ultra_feedback_dutch_cleaned](https://huggingface.co/datasets/BramVanroy/ultra_feedback_dutch_cleaned) subset `dpo_hq`: a cleaned version of [BramVanroy/ultra_feedback_dutch](https://huggingface.co/datasets/BramVanroy/ultra_feedback_dutch) (9186)
 - [BramVanroy/orca_dpo_pairs_dutch_cleaned](https://huggingface.co/datasets/BramVanroy/orca_dpo_pairs_dutch_cleaned) subset `dpo_all`: a cleaned version of [BramVanroy/orca_dpo_pairs_dutch](https://huggingface.co/datasets/BramVanroy/orca_dpo_pairs_dutch) (9467)
 
-A lot of different learning rates, beta, en batch sizes were investigated in search of a converging combination. You can find them all in [the W&B runs](https://wandb.ai/bramvanroy/dpo-fietje-2b).
+A lot of different learning rates, beta, en batch sizes were investigated in search of a converging combination. You can find them all in [the W&B runs](https://wandb.ai/bramvanroy/dpo-fietje-2).
 
 ## Training procedure