alfiannajih commited on
Commit
3043d12
1 Parent(s): dee4a24

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -1
config.json CHANGED
@@ -18,9 +18,16 @@
18
  "bos_token_id": 128000,
19
  "custom_pipelines": {
20
  "g-retriever-task": {
 
 
 
 
 
 
 
21
  "impl": "g_retriever_pipeline.GRetrieverPipeline",
22
  "pt": [
23
- "g_retriever_model.GRetrieverModel"
24
  ],
25
  "tf": []
26
  }
 
18
  "bos_token_id": 128000,
19
  "custom_pipelines": {
20
  "g-retriever-task": {
21
+ "default": {
22
+ "model": {
23
+ "pt": [
24
+ "alfiannajih/g-retriever",
25
+ "dee4a24"
26
+ ]
27
+ }
28
  "impl": "g_retriever_pipeline.GRetrieverPipeline",
29
  "pt": [
30
+ "AutoModelForCausalLM"
31
  ],
32
  "tf": []
33
  }