1  # Copyright 2024 NVIDIA CORPORATION & AFFILIATES
 2  #
 3  # Licensed under the Apache License, Version 2.0 (the "License");
 4  # you may not use this file except in compliance with the License.
 5  # You may obtain a copy of the License at
 6  #
 7  #      http://www.apache.org/licenses/LICENSE-2.0
 8  #
 9  # Unless required by applicable law or agreed to in writing, software
10  # distributed under the License is distributed on an "AS IS" BASIS,
11  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  # See the License for the specific language governing permissions and
13  # limitations under the License.
14  #
15  # SPDX-License-Identifier: Apache-2.0
16  # This file is modified from https://github.com/haotian-liu/LLaVA/
17  
18  
19  CONTROLLER_HEART_BEAT_EXPIRATION = 30
20  WORKER_HEART_BEAT_INTERVAL = 15
21  
22  LOGDIR = "."
23  
24  
25  # Model Constants
26  IGNORE_INDEX = -100
27  X_TOKEN_INDEX = -200
28  X_TOKEN =        {'image': "<|image_token|>", 'video': "<|video_token|>"}
29  X_PATCH_TOKEN =  {'image': "<|image_patch|>", 'video': "<|video_patch|>"}
30  X_START_TOKEN =  {'image': "<|image_start|>", 'video': "<|video_start|>"}
31  X_END_TOKEN =    {'image': "<|image_end|>",   'video': "<|video_end|>"}