rename embedder

This commit is contained in:
Aaron Miller 2023-07-13 18:23:20 -04:00 committed by AT
parent ee4186d579
commit 15f1fe5445
3 changed files with 3 additions and 3 deletions

View File

@ -1,2 +1,2 @@
from .gpt4all import GPT4All, Embedder # noqa
from .gpt4all import GPT4All, Embed4All # noqa
from .pyllmodel import LLModel # noqa

View File

@ -15,7 +15,7 @@ from . import pyllmodel
# TODO: move to config
DEFAULT_MODEL_DIRECTORY = os.path.join(str(Path.home()), ".cache", "gpt4all").replace("\\", "\\\\")
class Embedder:
class Embed4All:
def __init__(
self
):

View File

@ -1,7 +1,7 @@
import sys
from io import StringIO
from gpt4all import GPT4All, Embedder
from gpt4all import GPT4All, Embed4All
import time
def test_inference():