3213a65d96
* Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * remove submodule * add mPLUG MiniGPT4 * Update Readme.md * Update Readme.md * Update Readme.md --------- Co-authored-by: Yuliang Liu <34134635+Yuliang-Liu@users.noreply.github.com>
31 lines
950 B
Python
31 lines
950 B
Python
"""
|
|
Copyright (c) 2022, salesforce.com, inc.
|
|
All rights reserved.
|
|
SPDX-License-Identifier: BSD-3-Clause
|
|
For full license text, see the LICENSE_Lavis file in the repo root or https://opensource.org/licenses/BSD-3-Clause
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
|
|
from omegaconf import OmegaConf
|
|
from minigpt4.common.registry import registry
|
|
|
|
from minigpt4.datasets.builders import *
|
|
from minigpt4.models import *
|
|
from minigpt4.processors import *
|
|
from minigpt4.tasks import *
|
|
|
|
|
|
root_dir = os.path.dirname(os.path.abspath(__file__))
|
|
default_cfg = OmegaConf.load(os.path.join(root_dir, "configs/default.yaml"))
|
|
|
|
registry.register_path("library_root", root_dir)
|
|
repo_root = os.path.join(root_dir, "..")
|
|
registry.register_path("repo_root", repo_root)
|
|
cache_root = os.path.join(repo_root, default_cfg.env.cache_root)
|
|
registry.register_path("cache_root", cache_root)
|
|
|
|
registry.register("MAX_INT", sys.maxsize)
|
|
registry.register("SPLIT_NAMES", ["train", "val", "test"])
|