Yanisadel commited on
Commit
36fb493
·
verified ·
1 Parent(s): 93927ba

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -0
config.json CHANGED
@@ -7,6 +7,15 @@
7
  "AutoModel": "chatNT.TorchMultiOmicsModel"
8
  },
9
  "bio_pad_token_id": 1,
 
 
 
 
 
 
 
 
 
10
  "english_pad_token_id": 2,
11
  "esm_config": {
12
  "add_bias_ffn": false,
 
7
  "AutoModel": "chatNT.TorchMultiOmicsModel"
8
  },
9
  "bio_pad_token_id": 1,
10
+ "custom_pipelines": {
11
+ "ChatNT-text-generation": {
12
+ "impl": "text_generation.TextGenerationPipeline",
13
+ "pt": [
14
+ "AutoModel"
15
+ ],
16
+ "tf": []
17
+ }
18
+ },
19
  "english_pad_token_id": 2,
20
  "esm_config": {
21
  "add_bias_ffn": false,