Skip to content

Commit

Permalink
docs: correct types for example (#5188)
Browse files Browse the repository at this point in the history
  • Loading branch information
NitzanShwartz authored Jan 18, 2025
1 parent 1f1d8d4 commit e20f4cb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/build-with-bentoml/iotypes.rst
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ Python's standard types such as strings, integers, floats, booleans, lists, and
class LanguageModel:
@bentoml.api
def generate(
self, prompt: int = Field(description="The prompt text"),
self, prompt: str = Field(description="The prompt text"),
temperature: float = Field(default=0.0, description="A sampling temperature between 0 and 2"),
max_tokens: int = Field(default=1000, description="max tokens to use"),
) -> Generator[str, None, None]:
Expand Down

0 comments on commit e20f4cb

Please sign in to comment.