Skip to content

Commit

Permalink
Update keybert/llm/_textgenerationinference.py
Browse files Browse the repository at this point in the history
Co-authored-by: Maarten Grootendorst <[email protected]>
  • Loading branch information
afuetterer and MaartenGr authored Jul 15, 2024
1 parent 01993b3 commit 98f5e03
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion keybert/llm/_textgenerationinference.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ def extract_keywords(
For example, it will create a nicer representation of
the candidate keywords, remove redundant keywords, or
shorten them depending on the input prompt.
inference_kwargs: TODO
inference_kwargs: kwargs for `InferenceClient.text_generation`. See: https://huggingface.co/docs/huggingface_hub/package_reference/inference_client
Returns:
all_keywords: All keywords for each document
Expand Down

0 comments on commit 98f5e03

Please sign in to comment.