Add return type for default __call__ method

This commit is contained in:
Andrei Betlen 2023-04-03 20:26:08 -04:00
parent 4aa349d777
commit caf3c0362b

View file

@ -480,7 +480,7 @@ class Llama:
repeat_penalty: float = 1.1,
top_k: int = 40,
stream: bool = False,
):
) -> Union[Completion, Iterator[CompletionChunk]]:
"""Generate text from a prompt.
Args: