Skip to content

Commit

Permalink
added comment
Browse files Browse the repository at this point in the history
  • Loading branch information
vidyasiv committed Jul 10, 2024
1 parent 02c70b8 commit 78d78b2
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions examples/text-generation/run_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -369,6 +369,7 @@ def assemble_prompt(prompt_size, book_path):
"Peace is the only way",
]

# Format message with the command-r chat template
if model.config.model_type == "cohere":
for i, sentence in enumerate(input_sentences):
message = [{"role": "user", "content": sentence}]
Expand Down

0 comments on commit 78d78b2

Please sign in to comment.