Update README.md

README.md

@@ -69,13 +69,13 @@ You are a helpful assistant with access to the following functions. You must use
     "type": "function",
     "function": {
       "name": "TextToImage",
-      "description": "This function able to creating, drawing, or illustrating an image from a text prompt.",
+      "description": "This function is able to create, draw, or illustrate an image from a text prompt.",
       "parameters": {
         "type": "object",
         "properties": {
           "prompt": {
             "type": "string",
-            "description": "The description of image that user wanto to create."
+            "description": "The description of image that the user wants to create."
           }
         },
         "required": [
@@ -88,7 +88,7 @@ You are a helpful assistant with access to the following functions. You must use
 """
 messages = [
     {'role': 'system', 'content': system_prompt},
-    {'role': "user", 'content': "Help me to generate a picture of Eiffel Tower in the night!"}
+    {'role': "user", 'content': "Please, generate a picture of the Eiffel Tower at night!"}
 ]
 
 inputs = tokenizer.apply_chat_template(
@@ -111,7 +111,7 @@ print(output)
   {
     "name": "TextToImage",
     "arguments": {
-      "prompt": "Eiffel Tower in the night"
+      "prompt": "Eiffel Tower at night."
     }
   }
 ]
@@ -125,12 +125,12 @@ print(output)
 * **Developed by**: [Stability AI](https://stability.ai/)
 * **Model type**: `StableLM 2 12B Chat` model is an auto-regressive language model based on the transformer decoder architecture.
 * **Language(s)**: English
-TODO: Check if we want to keep paper link since it's not mentioned in that paper.
+TODO: Check if we want to keep paper link since this model is not explictly mentioned in the paper.
 * **Paper**: [Stable LM 2 Chat Technical Report](https://drive.google.com/file/d/1JYJHszhS8EFChTbNAf8xmqhKjogWRrQF/view?usp=sharing)
 * **Library**: [Alignment Handbook](https://github.com/huggingface/alignment-handbook.git)
 * **Finetuned from model**: 
 * **License**: [StabilityAI Non-Commercial Research Community License](https://huggingface.co/stabilityai/stablelm-2-zephyr-1_6b/blob/main/LICENSE). If you want to use this model for your commercial products or purposes, please contact us [here](https://stability.ai/contact) to learn more.
-* **Contact**: For questions and comments about the model, please email `[email protected]`
+* **Contact**: For questions and comments about the model, please email `[email protected]`.
 
 ### Training Dataset
 
@@ -149,7 +149,7 @@ The dataset is comprised of a mixture of open datasets large-scale datasets avai
 - Internal Safety Dataset
 
 3. Preference Datasets:
-
+- TBD.
 
 ## Performance
 
@@ -203,4 +203,11 @@ Additionally, as each use case is unique, we recommend running your own suite of
 Finally, do not use the models if they are unsuitable for your application, or for any applications that may cause deliberate or unintentional harm to others.
 
 
-## How to Cite
+## How to Cite
+
+@article{bellagente2024stable,
+  title={Stable LM 2 1.6 B Technical Report},
+  author={Bellagente, Marco and Tow, Jonathan and Mahan, Dakota and Phung, Duy and Zhuravinskyi, Maksym and Adithyan, Reshinth and Baicoianu, James and Brooks, Ben and Cooper, Nathan and Datta, Ashish and others},
+  journal={arXiv preprint arXiv:2402.17834},
+  year={2024}
+}