yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-07-16 21:40:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

merge lazy import blocks (#6358)

Browse Source
This commit is contained in:
Stefano Fiorucci 2023-11-21 11:15:37 +01:00 committed by GitHub
parent 2943b83b31
commit 1fff2bc255
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
haystack/preview/components/generators/hugging_face_local.py
View File

@ -9,12 +9,8 @@ logger = logging.getLogger(__name__)
SUPPORTED_TASKS = ["text-generation", "text2text-generation"]
with LazyImport(
message="PyTorch is needed to run this component. Please install it by following the instructions at https://pytorch.org/"
) as torch_import:
with LazyImport(message="Run 'pip install transformers[torch]'") as torch_and_transformers_import:
import torch
with LazyImport(message="Run 'pip install transformers'") as transformers_import:
from huggingface_hub import model_info
from transformers import (
pipeline,
@ -127,8 +123,7 @@ class HuggingFaceLocalGenerator:
For some chat models, the output includes both the new text and the original prompt.
In these cases, it's important to make sure your prompt has no stop words.
"""
transformers_import.check()
torch_import.check()
torch_and_transformers_import.check()
pipeline_kwargs = pipeline_kwargs or {}
generation_kwargs = generation_kwargs or {}