| """ | |
| Copyright (c) 2022, salesforce.com, inc. | |
| All rights reserved. | |
| SPDX-License-Identifier: BSD-3-Clause | |
| For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause | |
| """ | |
| import os | |
| import sys | |
| from omegaconf import OmegaConf | |
| from lavis.common.registry import registry | |
| from lavis.datasets.builders import * | |
| from lavis.models import * | |
| from lavis.processors import * | |
| from lavis.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"]) | |