Update README.md

README.md

@@ -11,10 +11,11 @@ model-index:
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
-
+# Cite 
+This model is trained from the code in this [GitHub](https://github.com/ShawhinT/YouTube-Blog/tree/main/LLMs/qlora)
 # shakespeare-ft
 
-This model is a fine-tuned version of [TheBloke/Mistral-7B-Instruct-v0.2-GPTQ](https://huggingface.co/TheBloke/Mistral-7B-Instruct-v0.2-GPTQ) on an unknown dataset.
+This model is a fine-tuned version of [TheBloke/Mistral-7B-Instruct-v0.2-GPTQ](https://huggingface.co/TheBloke/Mistral-7B-Instruct-v0.2-GPTQ) on an [Lambent/shakespeare_sonnets_backtranslated](https://huggingface.co/datasets/Lambent/shakespeare_sonnets_backtranslated) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.7122