chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-06 19:10:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix an import bug in image_text_pair_builder.py

Browse Source
This commit is contained in:
Deyao Zhu 2023-04-17 16:57:20 +03:00
parent b76d5c5a50
commit 3c9d321b79
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
minigpt4/common/config.py
View File

@ -71,7 +71,7 @@ class Config:
model_config_path = model_cls.default_config_path(model_type=model_type) model_config_path = model_cls.default_config_path(model_type=model_type)
model_config = OmegaConf.create() model_config = OmegaConf.create()
# hiararchy override, customized config > default config # hierarchy override, customized config > default config
model_config = OmegaConf.merge( model_config = OmegaConf.merge(
model_config, model_config,
OmegaConf.load(model_config_path), OmegaConf.load(model_config_path),
@ -102,7 +102,7 @@ class Config:
type=dataset_config_type type=dataset_config_type
) )
# hiararchy override, customized config > default config # hierarchy override, customized config > default config
dataset_config = OmegaConf.merge( dataset_config = OmegaConf.merge(
dataset_config, dataset_config,
OmegaConf.load(dataset_config_path), OmegaConf.load(dataset_config_path),

3
minigpt4/datasets/builders/image_text_pair_builder.py
View File

@ -1,5 +1,6 @@
import os import os
import logging import logging
import warnings
from minigpt4.common.registry import registry from minigpt4.common.registry import registry
from minigpt4.datasets.builders.base_dataset_builder import BaseDatasetBuilder from minigpt4.datasets.builders.base_dataset_builder import BaseDatasetBuilder
@ -90,7 +91,7 @@ class CCSBUAlignBuilder(BaseDatasetBuilder):
datasets = dict() datasets = dict()
if not os.path.exists(storage_path): if not os.path.exists(storage_path):
warnings.warn("storage path {} does not exist.".format(vis_path)) warnings.warn("storage path {} does not exist.".format(storage_path))
# create datasets # create datasets
dataset_cls = self.train_dataset_cls dataset_cls = self.train_dataset_cls