h2oai
/

h2ogpt-oasst1-512-12b

Text Generation

large language model

text-generation-inference

Model card Files Files and versions

arnocandel commited on Apr 18, 2023

Commit

a7828dc

·

1 Parent(s): 8962383

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -6,7 +6,7 @@
   "custom_pipelines": {
     "text-generation": {
       "impl": "h2oai_pipeline.H2OTextGenerationPipeline",
-      "pt": "AutoModelForCausalLM",
     }
   },
   "bos_token_id": 0,

   "custom_pipelines": {
     "text-generation": {
       "impl": "h2oai_pipeline.H2OTextGenerationPipeline",
+      "pt": "AutoModelForCausalLM"
     }
   },
   "bos_token_id": 0,