changing readme

README.md

@@ -1,3 +1,27 @@
 ---
 license: apache-2.0
+language:
+  - en
+pipeline_tag: text-generation
+inference: true
+tags:
+  - pytorch
+  - mistral
+  - finetuned
 ---
+# Mistral 7B - Holodeck
+## Model Description
+Mistral 7B-Holodeck is a finetune created using Mistral's 7B model.
+## Training data
+The training data contains around 3000 ebooks in various genres.
+Most parts of the dataset have been prepended using the following text: `[Genre: <genre1>, <genre2>]`
+### How to use
+You can use this model directly with a pipeline for text generation. This example generates a different sequence each time it's run:
+```py
+>>> from transformers import pipeline
+>>> generator = pipeline('text-generation', model='KoboldAI/Mistral-7B-Holodeck-1')
+>>> generator("Welcome Captain Janeway, I apologize for the delay.", do_sample=True, min_length=50)
+[{'generated_text': 'Welcome Captain Janeway, I apologize for the delay."\nIt's all right," Janeway said. "I'm certain that you're doing your best to keep me informed of what\'s going on."'}]
+```
+### Limitations and Biases
+Based on known problems with NLP technology, potential relevant factors include bias (gender, profession, race and religion).

config.json

@@ -22,5 +22,7 @@
   "torch_dtype": "float16",
   "transformers_version": "4.36.0",
   "use_cache": false,
-  "vocab_size": 32000
+  "vocab_size": 32000,
+  "welcome": "You are currently running story-writing model `Holodeck, version 1 (Mistral 7B).`\n\n This model is made by [Mr. Seeker](https://www.patreon.com/mrseeker) with help of the KoboldAI team.\n\n### How to use this model\n\Holodeck is designed to generate short stories and novels. Use the author's note to give it a certain genre to follow, use memory to give an overview of the story and use World Information to give specific details about the characters. To start off, give the AI an idea of what you are writing about by setting the scene. Give the AI around 10 sentences that make your story interesting to read. Introduce your character, describe the world, blow something up, or let the AI use its creative mind.",
+  "antemplate": "[Genre: <|>]"
 }