h2oai
/

h2ogpt-gm-oasst1-en-2048-falcon-7b-v2

Text Generation

RefinedWebModel

large language model

text-generation-inference

Model card Files Files and versions Community

ilu000 commited on Jun 2, 2023

Commit

bc9b92e

•

1 Parent(s): 3d57281

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -20,7 +20,7 @@
   "custom_pipelines": {
     "text-generation": {
       "impl": "h2oai_pipeline.H2OTextGenerationPipeline",
-      "pt": "modelling_RW.RWForCausalLM"
     }
   },
   "eos_token_id": 11,

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