VITA-1.5 / vita /constants.py
lxysl's picture
upload vita-1.5 app.py
bc752b1
raw
history blame
426 Bytes
# Model Constants
MAX_IMAGE_LENGTH = 16 # 8#16#32#64
MIN_IMAGE_LENGTH = 4
IGNORE_INDEX = -100
IMAGE_TOKEN_INDEX = -200
AUDIO_TOKEN_INDEX = -500
DEFAULT_IMAGE_TOKEN = "<image>"
DEFAULT_VIDEO_TOKEN = "<video>"
DEFAULT_AUDIO_TOKEN = "<audio>"
CONTROLLER_HEART_BEAT_EXPIRATION = 30
LOGDIR = "gradio-logs"
WORKER_HEART_BEAT_INTERVAL = 15
DEFAULT_DATA_RATIO = 1.0#0.124#0.5 #0.2 #1.0
GLOBAL_WEIGHTS_PATH = "/path/to/model_weights"