diff --git a/g4f/Provider/Local.py b/g4f/Provider/Local.py index c08064b9..471231c6 100644 --- a/g4f/Provider/Local.py +++ b/g4f/Provider/Local.py @@ -4,7 +4,7 @@ from ..locals.models import get_models try: from ..locals.provider import LocalProvider has_requirements = True -except ModuleNotFoundError: +except ImportError: has_requirements = False from ..typing import Messages, CreateResult @@ -12,7 +12,7 @@ from ..providers.base_provider import AbstractProvider, ProviderModelMixin from ..errors import MissingRequirementsError class Local(AbstractProvider, ProviderModelMixin): - label = "gpt4all" + label = "GPT4All" working = True supports_message_history = True supports_system_message = True