feat: HuggingFaceLocalGenerator - first implementation
#6022
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
github-actions
bot
added
topic:tests
type:documentation
anakin87
requested review from
dfokina and
vblagoje
and removed request for
a team
vblagoje
approved these changes
Oct 12, 2023
haystack/preview/components/generators/hugging_face/hugging_face_local.py
Outdated
Show resolved
Hide resolved
haystack/preview/components/generators/hugging_face/hugging_face_local.py
Show resolved
Hide resolved
haystack/preview/components/generators/hugging_face/hugging_face_local.py
Outdated
Show resolved
Hide resolved
|
@dfokina can you take a look at the docstrings? |
anakin87
changed the title
HuggingFaceLocalGeneratorHuggingFaceLocalGenerator - first implementation
dfokina
approved these changes
Oct 13, 2023
|
Sorry for extra spaces in docstrings @anakin87 😬 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Related Issues
HuggingFaceLocalGenerator- minimal implementation #6004Proposed Changes:
HuggingFaceLocalGeneratoris a 2.x component that generates text, by locally running a Hugging Face pipeline(same purpose of Hugging Face invocation layer for PromptNode)
Simple usage: you can just specify
model_name_or_pathtext generation can be influenced using
generation_kwargsCustomizable: experienced users can have full control of the Hugging Face pipeline using
pipeline_kwargs📓 Notebook with usage examples (potentially useful for docs and devrel purposes)
How did you test it?
CI, new unit tests, manual tests
Notes for the reviewer
This is a first implementation.
Some features (max length handling, stopwords, streaming) are missing and are tracked in #5953.
Checklist
fix:,feat:,build:,chore:,ci:,docs:,style:,refactor:,perf:,test:.