cmake_minimum_required(VERSION 3.16)

project(kylin-speech-asr-model)

add_subdirectory(
    punc_model_cpp/
)

install(DIRECTORY model_repo_paraformer_offline/
        DESTINATION /usr/share/kylin-ai/model-repository
        COMPONENT asr-model
        PATTERN ".gitkeep" EXCLUDE
        PATTERN "*~" EXCLUDE
        PATTERN "__pycache__" EXCLUDE)

install(DIRECTORY model_repo_paraformer_online/
        DESTINATION /usr/share/kylin-ai/model-repository
        COMPONENT asr-model
        PATTERN ".gitkeep" EXCLUDE
        PATTERN "*~" EXCLUDE
        PATTERN "__pycache__" EXCLUDE)

install(DIRECTORY model_repo_punc_dynamic_batching_small/
        DESTINATION /usr/share/kylin-ai/model-repository
        COMPONENT asr-model
        PATTERN ".gitkeep" EXCLUDE
        PATTERN "*~" EXCLUDE
        PATTERN "__pycache__" EXCLUDE)

install(
    TARGETS triton-punc-backend
    LIBRARY DESTINATION /usr/share/kylin-ai/model-repository/ensemble-punc_ct-transformer_nonstreaming/1/
)

install(DIRECTORY model_repo_vad_dynamic_batching/
        DESTINATION /usr/share/kylin-ai/model-repository
        COMPONENT asr-model
        PATTERN ".gitkeep" EXCLUDE
        PATTERN "*~" EXCLUDE
        PATTERN "__pycache__" EXCLUDE)
