From 53a1330ff37466871a64a9db3c3afcea61a19496 Mon Sep 17 00:00:00 2001 From: Matthew Hayes Date: Thu, 13 Apr 2023 03:17:10 +0000 Subject: [PATCH] Update config.json Try changing custom pipeline to `text-generation` instead to see if langchain will work with it. --- config.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.json b/config.json index cbec935..a089335 100755 --- a/config.json +++ b/config.json @@ -4,7 +4,7 @@ "GPTNeoXForCausalLM" ], "custom_pipelines": { - "instruction-following": { + "text-generation": { "impl": "instruct_pipeline.InstructionTextGenerationPipeline", "pt": "AutoModelForCausalLM", "tf": "TFAutoModelForCausalLM"