Skip to content

Commit 51a0a7b

Browse files
authored
[Text Gen UX] Add prompt as input param alias (#1270)
from internal specs: ```python text_pipeline = TextGeneration(model=MODEL, **kwargs) text_result = text_pipeline(prompt=PROMPT, **kwargs) ```
1 parent 3d9622b commit 51a0a7b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Diff for: src/deepsparse/transformers/pipelines/text_generation.py

+1
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ class Config:
8383
arbitrary_types_allowed = True
8484

8585
sequences: Union[str, List[str]] = Field(
86+
alias="prompt",
8687
description="The input sequences to generate the text from.",
8788
)
8889
include_prompt_logits: bool = Field(

0 commit comments

Comments
 (0)