yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-08-28 02:16:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move import to lazy import section (#6778)

Browse Source
This commit is contained in:
Vladimir Blagojevic 2024-01-18 17:34:55 +01:00 committed by GitHub
parent 0b177b3bc6
commit 40a8b2b4a9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
haystack/components/generators/chat/hugging_face_local.py
View File

@ -5,7 +5,6 @@ from typing import Any, Dict, List, Literal, Optional, Union, Callable
from haystack.components.generators.hf_utils import PIPELINE_SUPPORTED_TASKS from haystack.components.generators.hf_utils import PIPELINE_SUPPORTED_TASKS
from haystack import component, default_to_dict, default_from_dict from haystack import component, default_to_dict, default_from_dict
from haystack.components.generators.hf_utils import HFTokenStreamingHandler
from haystack.components.generators.utils import serialize_callback_handler, deserialize_callback_handler from haystack.components.generators.utils import serialize_callback_handler, deserialize_callback_handler
from haystack.dataclasses import ChatMessage, StreamingChunk from haystack.dataclasses import ChatMessage, StreamingChunk
from haystack.lazy_imports import LazyImport from haystack.lazy_imports import LazyImport
@ -16,7 +15,10 @@ logger = logging.getLogger(__name__)
with LazyImport(message="Run 'pip install transformers[torch]'") as torch_and_transformers_import: with LazyImport(message="Run 'pip install transformers[torch]'") as torch_and_transformers_import:
from huggingface_hub import model_info from huggingface_hub import model_info
from transformers import StoppingCriteriaList, pipeline, PreTrainedTokenizer, PreTrainedTokenizerFast from transformers import StoppingCriteriaList, pipeline, PreTrainedTokenizer, PreTrainedTokenizerFast
from haystack.components.generators.hf_utils import StopWordsCriteria # pylint: disable=ungrouped-imports from haystack.components.generators.hf_utils import ( # pylint: disable=ungrouped-imports
StopWordsCriteria,
HFTokenStreamingHandler,
)
from haystack.utils.hf import serialize_hf_model_kwargs, deserialize_hf_model_kwargs from haystack.utils.hf import serialize_hf_model_kwargs, deserialize_hf_model_kwargs