| 12345678910111213141516171819 |
- # -------------------------------------------------------------------------
- # Copyright (c) 2021-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
- #
- # NVIDIA CORPORATION & AFFILIATES and its licensors retain all intellectual
- # property and proprietary rights in and to this software, related
- # documentation and any modifications thereto. Any use, reproduction,
- # disclosure or distribution of this software and related documentation
- # without an express license agreement from NVIDIA CORPORATION is strictly
- # prohibited.
- #
- # Written by Jiarui Xu
- # -------------------------------------------------------------------------
- from .builder import build_model
- from .group_vit import GroupViT
- from .multi_label_contrastive import MultiLabelContrastive
- from .transformer import TextTransformer
- __all__ = ['build_model', 'MultiLabelContrastive', 'GroupViT', 'TextTransformer']
|
