chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 02:20:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

update config files

Browse Source
This commit is contained in:
junchen14 2023-11-02 02:07:07 +03:00
parent 748d18b5df
commit d057192f67
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
minigpt4/common/config.py
View File

@ -116,13 +116,14 @@ class Config:
@staticmethod @staticmethod
def build_evaluation_dataset_config(config): def build_evaluation_dataset_config(config):
datasets = config.get("evaluation_datasets", None) datasets = config.get("evaluation_datasets", None)
if datasets is None: # if datasets is None:
raise KeyError( # raise KeyError(
"Expecting 'datasets' as the root key for dataset configuration." # "Expecting 'datasets' as the root key for dataset configuration."
) # )
dataset_config = OmegaConf.create() dataset_config = OmegaConf.create()
if datasets is not None:
for dataset_name in datasets: for dataset_name in datasets:
builder_cls = registry.get_builder_class(dataset_name) builder_cls = registry.get_builder_class(dataset_name)