Complete readme. Fixed reference in gpt4all_j wrapper

This commit is contained in:
Iván martínez 2023-05-02 20:22:04 +02:00 committed by Iván Martínez
parent 55338b8f6e
commit 77447e50c0
2 changed files with 66 additions and 3 deletions

View file

@ -1,4 +1,4 @@
"""Wrapper for the GPT4All model."""
"""Wrapper for the GPT4All-J model."""
from functools import partial
from typing import Any, Dict, List, Mapping, Optional, Set
@ -117,7 +117,7 @@ class GPT4All_J(LLM):
**{
k: v
for k, v in self.__dict__.items()
if k in GPT4All._llama_param_names()
if k in GPT4All_J._llama_param_names()
},
}