Skip to content

Commit

Permalink
[Text Gen UX] Add prompt as input param alias (#1270)
Browse files Browse the repository at this point in the history
from internal specs:

```python
text_pipeline = TextGeneration(model=MODEL, **kwargs)
text_result = text_pipeline(prompt=PROMPT, **kwargs)
```
  • Loading branch information
bfineran committed Sep 25, 2023
1 parent 3d9622b commit 51a0a7b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/deepsparse/transformers/pipelines/text_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,7 @@ class Config:
arbitrary_types_allowed = True

sequences: Union[str, List[str]] = Field(
alias="prompt",
description="The input sequences to generate the text from.",
)
include_prompt_logits: bool = Field(
Expand Down

0 comments on commit 51a0a7b

Please sign in to comment.