matthayes commited on
Commit
53a1330
1 Parent(s): af3b4a2

Update config.json

Browse files

Try changing custom pipeline to `text-generation` instead to see if langchain will work with it.

Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -4,7 +4,7 @@
4
  "GPTNeoXForCausalLM"
5
  ],
6
  "custom_pipelines": {
7
- "instruction-following": {
8
  "impl": "instruct_pipeline.InstructionTextGenerationPipeline",
9
  "pt": "AutoModelForCausalLM",
10
  "tf": "TFAutoModelForCausalLM"
 
4
  "GPTNeoXForCausalLM"
5
  ],
6
  "custom_pipelines": {
7
+ "text-generation": {
8
  "impl": "instruct_pipeline.InstructionTextGenerationPipeline",
9
  "pt": "AutoModelForCausalLM",
10
  "tf": "TFAutoModelForCausalLM"