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 from .pyllmodel import LLModel # noqa

View File

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

View File

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