#!/usr/bin/env python3

# Copyright (c) Facebook, Inc. and its affiliates.
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
"""
Define PATHS to pre-trained models.
"""

RPA_RERANKER = 'zoo:light_whoami/rpa_reranker/model'
TEST_RPA_RERANKER = 'zoo:light_whoami/test_rpa_reranker/model'
RPA_RERANKER_AUTO_EXPANDED = (
    'zoo:light_whoami/rpa_reranker_auto_expanded_attention/model'
)
VANILLA_128 = 'zoo:light_whoami/vanilla_128/model'
VANILLA_1024 = 'zoo:light_whoami/vanilla_1024/model'
UL_128 = 'zoo:light_whoami/rpa_ul_128/model'
UL_1024 = 'zoo:light_whoami/rpa_ul_1024/model'
MULTIOBJECTIVE = 'zoo:light_whoami/multiobjective/model'
PROFILE_EXPANDED_128 = 'zoo:light_whoami/profile_expanded_attention_128/model'
PROFILE_EXPANDED_1024 = 'zoo:light_whoami/profile_expanded_attention_1024/model'
AUTO_EXPANDED_1024 = 'zoo:light_whoami/automated_expanded_attention_1024/model'
EXPANDED_AND_MULTIOBJECTIVE_1024 = (
    'zoo:light_whoami/expanded_and_multiobjective_1024/model'
)
