Update config.json

Try changing custom pipeline to `text-generation` instead to see if langchain will work with it.
This commit is contained in:
Matthew Hayes 2023-04-13 03:17:10 +00:00 committed by huggingface-web
parent af3b4a26a4
commit 53a1330ff3

@ -4,7 +4,7 @@
"GPTNeoXForCausalLM"
],
"custom_pipelines": {
"instruction-following": {
"text-generation": {
"impl": "instruct_pipeline.InstructionTextGenerationPipeline",
"pt": "AutoModelForCausalLM",
"tf": "TFAutoModelForCausalLM"