databricks/dolly-v2-12b
databricks/dolly-v2-12b
1
0
Overview Code Discussions Pull Requests Wiki Activity

Update config.json

Browse Source 
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
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.json
View File

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