commit files to HF hub

README.md

@@ -12,6 +12,7 @@ tags:
 - open-source
 datasets:
 - h2oai/openassistant_oasst1
+- h2oai/openassistant_oasst1_h2ogpt
 ---
 # h2oGPT Model Card
 ## Summary
@@ -19,9 +20,9 @@ datasets:
 H2O.ai's `h2ogpt-oasst1-512-12b` is a 12 billion parameter instruction-following large language model licensed for commercial use.
 
 - Base model: [EleutherAI/pythia-12b](https://huggingface.co/EleutherAI/pythia-12b)
-- Fine-tuning dataset: [h2oai/openassistant_oasst1](https://huggingface.co/datasets/h2oai/openassistant_oasst1)
+- Fine-tuning dataset: [h2oai/openassistant_oasst1](https://huggingface.co/datasets/h2oai/openassistant_oasst1) and [h2oai/openassistant_oasst1_h2ogpt](https://huggingface.co/datasets/h2oai/openassistant_oasst1_h2ogpt)
 - Data-prep and fine-tuning code: [H2O.ai GitHub](https://github.com/h2oai/h2ogpt)
-- Training logs: [zip](https://huggingface.co/h2oai/h2ogpt-oasst1-512-12b/blob/main/pythia-12b.openassistant_oasst1.json.1_epochs.d45a9d34d34534e076cc6797614b322bd0efb11c.15.zip)
+- Training logs: [zip](https://huggingface.co/h2oai/h2ogpt-oasst1-512-12b/blob/main/pythia-12b.openassistant_oasst1.json.1_epochs.d45a9d34d34534e076cc6797614b322bd0efb11c.15.zip) and [zip](https://huggingface.co/h2oai/h2ogpt-oasst1-512-12b/blob/main/h2ogpt-oasst1-512-12b.h2oaiopenassistant_oasst1_h2ogpt.2_epochs.fcaae7ef70600de8c97c9b38cb3f0075467cdad1.2.zip)
 
 ## Chatbot
 
@@ -100,6 +101,12 @@ GPTNeoXConfig {
     "GPTNeoXForCausalLM"
   ],
   "bos_token_id": 0,
+  "custom_pipeline": {
+    "text-generation": {
+      "impl": "h2oai_pipeline.H2OTextGenerationPipeline",
+      "pt": "AutoModelForCausalLM"
+    }
+  },
   "custom_pipelines": {
     "text-generation": {
       "impl": "h2oai_pipeline.H2OTextGenerationPipeline",