Correct return type

hotfix/python-download-model-bug
Konstantin Gukov 1 year ago committed by Richard Guo
parent 100c809f1e
commit e05ee9466a

@ -173,7 +173,7 @@ class GPT4All():
default_prompt_footer: bool = True,
verbose: bool = True,
streaming: bool = True,
**generate_kwargs) -> str:
**generate_kwargs) -> dict:
"""
Format list of message dictionaries into a prompt and call model
generate on prompt. Returns a response dictionary with metadata and

Loading…
Cancel
Save