Update README.md

README.md

@@ -9,6 +9,7 @@ tags:
 - mistral
 - trl
 - sft
+- gguf
 base_model: unsloth/mistral-7b-v0.2-bnb-4bit
 datasets:
 - visheratin/realworldqa
@@ -18,19 +19,47 @@ datasets:
 <img src="https://i.imgur.com/Pf53ms5.jpeg" width="400"/>
 
 
-An experiment with the goal of reducing halucinations in [VQA](https://huggingface.co/tasks/visual-question-answering)
-First in of a series of projects centering around fine-tuning for image captioning.
+An experiment with the goal of reducing hallucinations in [VQA](https://huggingface.co/tasks/visual-question-answering)
+
+First in a series of experiments centering around fine-tuning for image captioning.
+
+<h1>Release Notes</h1>
 
-Release Notes:
 * v0.1 - Initial Release
-* v0.2 (Current)- Updating base model to official Mistral-7b fp16 release, refinements to dataset and instruction formating
+* <b>v0.2</b> (Current)- Updating base model to official Mistral-7b fp16 release, refinements to dataset and instruction formating
 
+<h2>Background & Methodology</h2>
 
 Mistral-7b-02 base model was fine-tuned using the [RealWorldQA dataset](https://huggingface.co/datasets/visheratin/realworldqa), originally provided by the X.Ai Team here: https://x.ai/blog/grok-1.5v
 
+<h1>Vision Results</h1>
+
+Example 1
+<img src="https://i.imgur.com/E9mS4Xb.jpeg" width="400"/>
+Example 2
+<img src="https://i.imgur.com/SmTz1Yd.jpeg" width="400"/>
+
+* Experiment yielded model that provides shorter, less verbose output for questions about pictures
+* The likelihood of hallucinations in output has decreased, however, the model can still be easily influenced to be inaccurate by the user
+* Best suited for captioning use cases that require concise descriptions and low token counts
+* This model lacks the conversational prose of Excalibur-7b-DPO and is much "drier" in tone
+
+<b>Requires additional mmproj file. You have two options for vision functionality (available inside this repo):</b>
+ 1. [Quantized - Limited VRAM Option (197mb)](https://huggingface.co/InferenceIllusionist/Mistral-RealworldQA-v0.2-7b-SFT/resolve/main/mistral-7b-mmproj-v1.5-Q4_1.gguf?download=true)
+ 2. [Unquantized - Premium Option / Best Quality (596mb)](https://huggingface.co/InferenceIllusionist/Mistral-RealworldQA-v0.2-7b-SFT/resolve/main/mmproj-model-f16.gguf?download=true)
+
+Select the gguf file of your choice in [Koboldcpp](https://github.com/LostRuins/koboldcpp/releases/) as usual, then make sure to choose the mmproj file above in the LLaVA mmproj field of the model submenu:
+<img src="https://i.imgur.com/x8vqH29.png" width="425"/>
+
+## Prompt Format
+Use Alpaca for best results.
+
+
+## Other info
 - **Developed by:** InferenceIllusionist
 - **License:** apache-2.0
 - **Finetuned from model :** mistral-community/Mistral-7B-v0.2
 
+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)
+[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)