Skip to content

Commit

Permalink
Update ram/data_utils.py
Browse files Browse the repository at this point in the history
Co-authored-by: Golovneva <103262907+Golovneva@users.noreply.github.com>
  • Loading branch information
uralik and Golovneva authored Sep 26, 2024
1 parent 3666f35 commit e36a20c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ram/data_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -481,7 +481,7 @@ def wrap_text_with_chat_template(
Args:
text (str): text of the user prompt
tokenizer (AutoTokenizer): Instantiated tokenizer.
Example: AutoTokenizer.from_pretrained("/fsx-ram/shared/llama2_hf/llama-2-7b-chat-hf/")
Example: AutoTokenizer.from_pretrained("<local_path>")
system_prompt (str): added following the model if tokenizer supports that, otherwise the error will be returned. Default is None.
"""

Expand Down

0 comments on commit e36a20c

Please sign in to comment.