yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-12-28 23:48:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Fix lazy import in HuggingFaceLocalGenerator (#6752)

Browse Source 
* Fix lazy import in HuggingFaceLocalGenerator

* Fix pylint

* Import fix after merge
This commit is contained in:
Vladimir Blagojevic 2024-01-17 14:32:03 +01:00 committed by GitHub
parent 7376838922
commit 88191e74bf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
haystack/components/generators/hugging_face_local.py
View File

@ -2,7 +2,7 @@ import logging
from typing import Any, Dict, List, Literal, Optional, Union
from haystack import component, default_from_dict, default_to_dict
from haystack.components.generators.hf_utils import StopWordsCriteria
from haystack.lazy_imports import LazyImport
from haystack.utils import ComponentDevice
@ -14,6 +14,7 @@ with LazyImport(message="Run 'pip install transformers[torch]'") as torch_and_tr
import torch
from huggingface_hub import model_info
from transformers import StoppingCriteriaList, pipeline
from haystack.components.generators.hf_utils import StopWordsCriteria # pylint: disable=ungrouped-imports
@component