Update dataset.py

Fix typo in a parameter name.
This commit is contained in:
Tianyu Yu
2024-10-22 16:08:52 +08:00
committed by GitHub
parent 2ab9fa7913
commit 8464c94a7b

View File

@@ -380,7 +380,7 @@ def preprocess(
) )
else: else:
conversations[0]["content"] = ( conversations[0]["content"] = (
image_placeholder + "\n" + conversation[0]["content"] image_placeholder + "\n" + conversations[0]["content"]
) )
input_dict = conversation_to_ids(conversations, tokenizer, llm_type, new_schema, max_length) input_dict = conversation_to_ids(conversations, tokenizer, llm_type, new_schema, max_length)
else: else: