Skip to content

Commit

Permalink
add prompt string to nvgpt
Browse files Browse the repository at this point in the history
Signed-off-by: HuiyingLi <[email protected]>
  • Loading branch information
HuiyingLi committed Feb 16, 2024
1 parent 7f1f85c commit 9b615b8
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions nemo/collections/multimodal/data/neva/neva_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -381,6 +381,8 @@ def preprocess_nvgpt(sources: dict, tokenizer, cfg,) -> Dict:
- The function asserts that each message in a conversation alternates between the defined roles and skips messages not starting with the 'human' role.
"""

"""<extra_id_0>System\nA chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.\n\n<extra_id_1>User\n{user input}\n<extra_id_1>Assistant\n<extra_id_2>quality:4,toxicity:0,humor:0,creativity:0,helpfulness:4,correctness:4,coherence:4,complexity:4,verbosity:4\n"""

conv = conversation_lib.conv_nvgpt.copy()

# Apply prompt templates
Expand Down

0 comments on commit 9b615b8

Please sign in to comment.