diff --git a/custom-models/bedrock-fine-tuning/nova/understanding/dataset_validation/nova_ft_dataset_validator.py b/custom-models/bedrock-fine-tuning/nova/understanding/dataset_validation/nova_ft_dataset_validator.py index c4489c9a8..4ac272d22 100644 --- a/custom-models/bedrock-fine-tuning/nova/understanding/dataset_validation/nova_ft_dataset_validator.py +++ b/custom-models/bedrock-fine-tuning/nova/understanding/dataset_validation/nova_ft_dataset_validator.py @@ -1,16 +1,30 @@ import argparse import json import re - -from pydantic import BaseModel, ValidationError, ValidationInfo, field_validator, model_validator from typing import List, Optional +from pydantic import BaseModel, ValidationError, ValidationInfo, field_validator, model_validator IMAGE_FORMATS = ["jpeg", "png", "gif", "webp"] VIDEO_FORMATS = ["mov", "mkv", "mp4", "webm"] MAX_NUM_IMAGES = 10 MODEL_TO_NUM_SAMPLES_MAP = {"micro": (8, 20000), "lite": (8, 20000), "pro": (8, 20000)} +INVALID_TOKENS_TEXT = [ + "System:", + "SYSTEM:", + "User:", + "USER:", + "Bot:", + "BOT:", + "Assistant:", + "ASSISTANT:", + "Thought:", + "[EOS]", + "", + "