psinger commited on
Commit
a2d3a4a
·
1 Parent(s): a72ae89

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -12
README.md CHANGED
@@ -8,7 +8,11 @@ tags:
8
  - large language model
9
  - h2o-llmstudio
10
  inference: false
11
- thumbnail: https://h2o.ai/etc.clientlibs/h2o/clientlibs/clientlib-site/resources/images/favicon.ico
 
 
 
 
12
  ---
13
  # Model Card
14
  ## Summary
@@ -22,7 +26,7 @@ This model was trained using [H2O LLM Studio](https://github.com/h2oai/h2o-llmst
22
  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.
23
 
24
  ```bash
25
- pip install transformers==4.30.0.dev0
26
  pip install accelerate==0.19.0
27
  pip install torch==2.0.0
28
  ```
@@ -173,16 +177,6 @@ LlamaForCausalLM(
173
 
174
  This model was trained using H2O LLM Studio and with the configuration in [cfg.yaml](cfg.yaml). Visit [H2O LLM Studio](https://github.com/h2oai/h2o-llmstudio) to learn how to train your own large language models.
175
 
176
-
177
- ## Model Validation
178
-
179
- Model validation results using [EleutherAI lm-evaluation-harness](https://github.com/EleutherAI/lm-evaluation-harness).
180
-
181
- ```bash
182
- CUDA_VISIBLE_DEVICES=0 python main.py --model hf-causal-experimental --model_args pretrained=h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-7b --tasks openbookqa,arc_easy,winogrande,hellaswag,arc_challenge,piqa,boolq --device cuda &> eval.log
183
- ```
184
-
185
-
186
  ## Disclaimer
187
 
188
  Please read this disclaimer carefully before using the large language model provided in this repository. Your use of the model signifies your agreement to the following terms and conditions.
 
8
  - large language model
9
  - h2o-llmstudio
10
  inference: false
11
+ thumbnail: >-
12
+ https://h2o.ai/etc.clientlibs/h2o/clientlibs/clientlib-site/resources/images/favicon.ico
13
+ license: apache-2.0
14
+ datasets:
15
+ - OpenAssistant/oasst1
16
  ---
17
  # Model Card
18
  ## Summary
 
26
  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.
27
 
28
  ```bash
29
+ pip install transformers==4.29.2
30
  pip install accelerate==0.19.0
31
  pip install torch==2.0.0
32
  ```
 
177
 
178
  This model was trained using H2O LLM Studio and with the configuration in [cfg.yaml](cfg.yaml). Visit [H2O LLM Studio](https://github.com/h2oai/h2o-llmstudio) to learn how to train your own large language models.
179
 
 
 
 
 
 
 
 
 
 
 
180
  ## Disclaimer
181
 
182
  Please read this disclaimer carefully before using the large language model provided in this repository. Your use of the model signifies your agreement to the following terms and conditions.