add einops

add details on usage/model constraints too

README.md

@@ -6,6 +6,10 @@ pipeline_tag: text-generation
 inference: false
 datasets:
 - the_pile_books3
+tags:
+- mosaicML
+- sharded
+- story
 ---
 
 # mpt-7b-storywriter: sharded
@@ -22,10 +26,12 @@ Please refer to the previously linked repo for details on usage/implementation/e
 
 ## Basic Usage
 
+> Note when using: this is **not** an instruction-tuned model, so you need to give it sufficient input text to continue generating something on-topic with your prompt
+> 
 Install/upgrade packages:
 
 ```bash
-pip install -U torch transformers accelerate
+pip install -U torch transformers accelerate einops
 ```
 
 Load the model:
@@ -50,5 +56,4 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 Then you can use `model.generate()` as you would normally - see the notebook for details.
 
 
----
-
+---