Update README.md

README.md

@@ -6,8 +6,27 @@ language:
 - en
 pipeline_tag: text-generation
 ---
-## Training procedure
 
+# ctrltokyo/llama-2-7b-hf-dolly-flash-attention
+
+This model is a fine-tuned version of [NousResearch/Llama-2-7b-hf](https://huggingface.co/NousResearch/Llama-2-7b-hf) on the databricks/databricks-dolly-15k dataset with all training performed using Flash Attention 2.
+
+No further testing or optimisation has been performed.
+
+## Model description
+
+Just like [ctrltokyo/llm_prompt_mask_fill_model](https://huggingface.co/ctrltokyo/llm_prompt_mask_fill_model), this model could be used for live autocompletion of PROMPTS, but is more designed for a generalized chatbot (hence the usage of the Dolly 15k dataset). Don't try this on code, because it won't work.
+I plan to release a further fine-tuned version using the [code_instructions_120k](https://huggingface.co/datasets/sahil2801/code_instructions_120k) dataset.
+
+## Intended uses & limitations
+
+Use as intended.
+
+## Training and evaluation data
+
+No evaluation was performed. Trained on NVIDIA A100, but appears to use around 20GB of VRAM when performing inference on the raw model.
+
+## Training procedure
 
 The following `bitsandbytes` quantization config was used during training:
 - load_in_8bit: False