diff --git a/models/config.py b/models/config.py index 08f74dbb..4639004a 100644 --- a/models/config.py +++ b/models/config.py @@ -72,6 +72,6 @@ class TrainConfig: wandb_entity: str = "HuggingFace" # Indicate the entity to log to in wandb log_wandb: bool = True use_lmms_eval: bool = True # Use lmms-eval for evaluation - lmms_eval_tasks: str = 'mmstar,mmmu,ocrbench,textvqa' # Pass additional task as one string, seperated by commas without spaces (e.g. 'mmstar,mmmu,ocrbench') + lmms_eval_tasks: str = 'mmstar,mmmu,ocrbench,textvqa' # Pass additional task as one string, separated by commas without spaces (e.g. 'mmstar,mmmu,ocrbench') lmms_eval_limit: int = 2000 - lmms_eval_batch_size: int = 128 \ No newline at end of file + lmms_eval_batch_size: int = 128