config.py

# *_*coding:utf-8 *_*
import os


PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))


def _path_from_env(env_name, default_relative_path):
    path = os.environ.get(env_name)
    if path:
        return os.path.expanduser(path)
    return os.path.join(PROJECT_ROOT, default_relative_path)


def _join(root, *parts):
    return os.path.join(root, *parts)


# Runtime roots - overridden by app.py at startup with HF Hub paths
MODEL_ROOT = _path_from_env("SWDH_MODEL_ROOT", "models")

PATH_TO_LLM = {
    'Qwen25': _join(MODEL_ROOT, 'Qwen2.5-7B-Instruct'),
    'Qwen3': _join(MODEL_ROOT, 'Qwen3-8B'),
    'Qwen3_0.6B': _join(MODEL_ROOT, 'Qwen3-0.6B'),
}

PATH_TO_VISUAL = {
    'CLIP_VIT_LARGE': _join(MODEL_ROOT, 'clip-vit-large-patch14'),
    'CLIP_VIT_BASE': _join(MODEL_ROOT, 'clip-vit-base-patch16'),
}

PATH_TO_AUDIO = {
    'HUBERT_LARGE': _join(MODEL_ROOT, 'chinese-hubert-large'),
    'HUBERT_BASE': _join(MODEL_ROOT, 'chinese-hubert-base'),
}


#######################
## store global values
#######################
DEFAULT_IMAGE_PATCH_TOKEN = '<ImageHere>'
DEFAULT_AUDIO_PATCH_TOKEN = '<AudioHere>'
DEFAULT_FRAME_PATCH_TOKEN = '<FrameHere>'
DEFAULT_FACE_PATCH_TOKEN = '<FaceHere>'
DEFAULT_MULTI_PATCH_TOKEN = '<MultiHere>'
IGNORE_INDEX = -100