Update README.md

README.md

@@ -8,13 +8,18 @@ tags:
 - large language model
 - h2o-llmstudio
 inference: false
-thumbnail: https://h2o.ai/etc.clientlibs/h2o/clientlibs/clientlib-site/resources/images/favicon.ico
+thumbnail: >-
+  https://h2o.ai/etc.clientlibs/h2o/clientlibs/clientlib-site/resources/images/favicon.ico
+license: apache-2.0
+datasets:
+- OpenAssistant/oasst1
 ---
 # Model Card
 ## Summary
 
 This model was trained using [H2O LLM Studio](https://github.com/h2oai/h2o-llmstudio).
 - Base model: [tiiuae/falcon-7b](https://huggingface.co/tiiuae/falcon-7b)
+- Dataset preparation: [OpenAssistant/oasst1](https://github.com/h2oai/h2o-llmstudio/blob/1935d84d9caafed3ee686ad2733eb02d2abfce57/app_utils/utils.py#LL1896C5-L1896C28)
 
 
 ## Usage
@@ -22,9 +27,10 @@ This model was trained using [H2O LLM Studio](https://github.com/h2oai/h2o-llmst
 To use the model with the `transformers` library on a machine with GPUs, first make sure you have the `transformers`, `accelerate` and `torch` libraries installed.
 
 ```bash
-pip install transformers==4.28.1
-pip install accelerate==0.18.0
+pip install transformers==4.29.2
+pip install accelerate==0.19.0
 pip install torch==2.0.0
+pip install einops==0.6.1
 ```
 
 ```python