arnocandel commited on
Commit
ecb5ad1
1 Parent(s): 36c9d9d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -0
config.json CHANGED
@@ -3,6 +3,12 @@
3
  "architectures": [
4
  "GPTNeoXForCausalLM"
5
  ],
 
 
 
 
 
 
6
  "bos_token_id": 0,
7
  "eos_token_id": 0,
8
  "hidden_act": "gelu",
 
3
  "architectures": [
4
  "GPTNeoXForCausalLM"
5
  ],
6
+ "custom_pipelines": {
7
+ "text-generation": {
8
+ "impl": "h2oai_pipeline.H2OTextGenerationPipeline",
9
+ "pt": "AutoModelForCausalLM",
10
+ }
11
+ },
12
  "bos_token_id": 0,
13
  "eos_token_id": 0,
14
  "hidden_act": "gelu",