h2oai
/

h2ogpt-oig-oasst1-256-6_9b

Text Generation

large language model

text-generation-inference

Model card Files Files and versions Community

arnocandel commited on Apr 18, 2023

Commit

cf67bde

•

1 Parent(s): 047640d

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,