matthayes commited on
Commit
43dda91
1 Parent(s): fb9c3bc

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -0
config.json CHANGED
@@ -3,6 +3,13 @@
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": "instruct_pipeline.InstructionTextGenerationPipeline",
9
+ "pt": "AutoModelForCausalLM",
10
+ "tf": "TFAutoModelForCausalLM"
11
+ }
12
+ },
13
  "bos_token_id": 0,
14
  "eos_token_id": 0,
15
  "hidden_act": "gelu",