jacobrenn commited on
Commit
6c55b32
1 Parent(s): c865215

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -0
config.json CHANGED
@@ -6,6 +6,13 @@
6
  "architectures": [
7
  "OPTForCausalLM"
8
  ],
 
 
 
 
 
 
 
9
  "attention_dropout": 0.0,
10
  "bos_token_id": 2,
11
  "do_layer_norm_before": true,
 
6
  "architectures": [
7
  "OPTForCausalLM"
8
  ],
9
+ "custom_pipelines": {
10
+ "text-generation": {
11
+ "impl": "instruct_pipeline.InstructionTextGenerationPipeline",
12
+ "pt": "AutoModelForCausalLM",
13
+ "tf": "TFAutoModelForCausalLM"
14
+ }
15
+ },
16
  "attention_dropout": 0.0,
17
  "bos_token_id": 2,
18
  "do_layer_norm_before": true,