custom_oneformer/__init__.py

# Copyright (c) Facebook, Inc. and its affiliates.
from . import data  # register all new datasets
from . import modeling

# config
from .config import *

# models
from .oneformer_model import OneFormer