diff --git a/example_chat_completion.py b/example_chat_completion.py index 712dbac..4f8373d 100644 --- a/example_chat_completion.py +++ b/example_chat_completion.py @@ -18,7 +18,7 @@ def main( max_gen_len: Optional[int] = None, ): """ - Examples to run with the models finetuned for chat. Prompts correspond of chat + Examples to run with the models finetuned for chat. Prompts correspond to chat turns between the user and assistant with the final one always being the user. An optional system prompt at the beginning to control how the model should respond @@ -82,3 +82,4 @@ def main( if __name__ == "__main__": fire.Fire(main) +}