mirror of
https://github.com/immich-app/ml-models.git
synced 2026-09-30 13:22:55 +08:00
60e028dee6
* feat: add ppocr to exporter * ocr: unpack scrfd heads * ocr: affine fold Co-authored-by: todorangrg <todorangrg@gmail.com> * rknn dynamic inputs * fp16 outputs * tiled transpose * refactor * target multiple aspect ratios * update cli * add 16:9 * better ci fixture integration * ci bug fixes * check cancellation * chore: bless ci fixtures * tweak runners * add metadata to rknn binaries * canvas->dim * fix fixture trigger * update onnxscript locals * chore: bless ci fixtures * download unconditionally * simplify fix-fixtures * use huge runner for server model * run compilation in subprocess * try/catch recv * remove 16/9 * fix detection mac utilization * chore: bless ci fixtures * conditional rknn compile * add rknn config to render --------- Co-authored-by: todorangrg <todorangrg@gmail.com> Co-authored-by: github-actions <41898282+github-actions[bot]@users.noreply.github.com>
175 lines
5.1 KiB
YAML
175 lines
5.1 KiB
YAML
models:
|
|
- name: 'LABSE-Vit-L-14'
|
|
hf-name: 'M-CLIP/LABSE-Vit-L-14'
|
|
source: 'mclip'
|
|
- name: 'XLM-Roberta-Large-Vit-B-16Plus'
|
|
hf-name: 'M-CLIP/XLM-Roberta-Large-Vit-B-16Plus'
|
|
source: 'mclip'
|
|
- name: 'XLM-Roberta-Large-Vit-B-32'
|
|
hf-name: 'M-CLIP/XLM-Roberta-Large-Vit-B-32'
|
|
source: 'mclip'
|
|
- name: 'XLM-Roberta-Large-Vit-L-14'
|
|
hf-name: 'M-CLIP/XLM-Roberta-Large-Vit-L-14'
|
|
source: 'mclip'
|
|
- name: 'PP-OCRv5_mobile'
|
|
source: 'paddle'
|
|
- name: 'PP-OCRv5_server'
|
|
source: 'paddle'
|
|
runner: 'pokedex-huge'
|
|
- name: 'CH__PP-OCRv5_mobile'
|
|
source: 'paddle'
|
|
- name: 'CH__PP-OCRv5_server'
|
|
source: 'paddle'
|
|
runner: 'pokedex-huge'
|
|
- name: 'EL__PP-OCRv5_mobile'
|
|
source: 'paddle'
|
|
- name: 'EN__PP-OCRv5_mobile'
|
|
source: 'paddle'
|
|
- name: 'ESLAV__PP-OCRv5_mobile'
|
|
source: 'paddle'
|
|
- name: 'KOREAN__PP-OCRv5_mobile'
|
|
source: 'paddle'
|
|
- name: 'LATIN__PP-OCRv5_mobile'
|
|
source: 'paddle'
|
|
- name: 'TH__PP-OCRv5_mobile'
|
|
source: 'paddle'
|
|
- name: 'PP-OCRv6_tiny'
|
|
source: 'paddle'
|
|
- name: 'PP-OCRv6_small'
|
|
source: 'paddle'
|
|
- name: 'PP-OCRv6_medium'
|
|
source: 'paddle'
|
|
runner: 'pokedex-large'
|
|
- name: 'antelopev2'
|
|
source: 'insightface'
|
|
- name: 'buffalo_l'
|
|
source: 'insightface'
|
|
- name: 'buffalo_m'
|
|
source: 'insightface'
|
|
- name: 'buffalo_s'
|
|
source: 'insightface'
|
|
- name: 'RN101__openai'
|
|
source: 'openclip'
|
|
- name: 'RN101__yfcc15m'
|
|
source: 'openclip'
|
|
- name: 'RN50__cc12m'
|
|
source: 'openclip'
|
|
- name: 'RN50__openai'
|
|
source: 'openclip'
|
|
- name: 'RN50__yfcc15m'
|
|
source: 'openclip'
|
|
- name: 'RN50x16__openai'
|
|
source: 'openclip'
|
|
- name: 'RN50x4__openai'
|
|
source: 'openclip'
|
|
- name: 'RN50x64__openai'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16-SigLIP-256__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16-SigLIP-384__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16-SigLIP-512__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16-SigLIP-i18n-256__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16-SigLIP2__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16-SigLIP__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16-plus-240__laion400m_e31'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16-plus-240__laion400m_e32'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16__laion400m_e31'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16__laion400m_e32'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-16__openai'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-32-SigLIP2-256__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-32__laion2b-s34b-b79k'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-32__laion2b_e16'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-32__laion400m_e31'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-32__laion400m_e32'
|
|
source: 'openclip'
|
|
- name: 'ViT-B-32__openai'
|
|
source: 'openclip'
|
|
- name: 'ViT-H-14-378-quickgelu__dfn5b'
|
|
source: 'openclip'
|
|
runner: 'pokedex-large'
|
|
- name: 'ViT-H-14-quickgelu__dfn5b'
|
|
source: 'openclip'
|
|
runner: 'pokedex-medium'
|
|
- name: 'ViT-H-14__laion2b-s32b-b79k'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-14-336__openai'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-14-quickgelu__dfn2b'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-14__laion2b-s32b-b82k'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-14__laion400m_e31'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-14__laion400m_e32'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-14__openai'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-16-SigLIP-256__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-16-SigLIP-384__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-16-SigLIP2-256__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-16-SigLIP2-384__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-L-16-SigLIP2-512__webli'
|
|
source: 'openclip'
|
|
- name: 'ViT-SO400M-14-SigLIP-384__webli'
|
|
source: 'openclip'
|
|
runner: 'pokedex-large'
|
|
- name: 'ViT-SO400M-14-SigLIP2-378__webli'
|
|
source: 'openclip'
|
|
runner: 'pokedex-large'
|
|
- name: 'ViT-SO400M-14-SigLIP2__webli'
|
|
source: 'openclip'
|
|
runner: 'pokedex-large'
|
|
- name: 'ViT-SO400M-16-SigLIP2-256__webli'
|
|
source: 'openclip'
|
|
runner: 'pokedex-medium'
|
|
- name: 'ViT-SO400M-16-SigLIP2-384__webli'
|
|
source: 'openclip'
|
|
runner: 'pokedex-medium'
|
|
- name: 'ViT-SO400M-16-SigLIP2-512__webli'
|
|
source: 'openclip'
|
|
runner: 'pokedex-huge'
|
|
- name: 'ViT-gopt-16-SigLIP2-256__webli'
|
|
source: 'openclip'
|
|
runner: 'pokedex-huge'
|
|
- name: 'ViT-gopt-16-SigLIP2-384__webli'
|
|
source: 'openclip'
|
|
runner: 'pokedex-huge'
|
|
- name: 'nllb-clip-base-siglip__mrl'
|
|
source: 'openclip'
|
|
runner: 'pokedex-medium'
|
|
- name: 'nllb-clip-base-siglip__v1'
|
|
source: 'openclip'
|
|
runner: 'pokedex-medium'
|
|
- name: 'nllb-clip-large-siglip__mrl'
|
|
source: 'openclip'
|
|
runner: 'pokedex-huge'
|
|
- name: 'nllb-clip-large-siglip__v1'
|
|
source: 'openclip'
|
|
runner: 'pokedex-huge'
|
|
- name: 'XLM-Roberta-Base-ViT-B-32__laion5b_s13b_b90k'
|
|
hf-name: 'xlm-roberta-base-ViT-B-32__laion5b_s13b_b90k'
|
|
source: 'openclip'
|
|
runner: 'pokedex-medium'
|
|
- name: 'XLM-Roberta-Large-ViT-H-14__frozen_laion5b_s13b_b90k'
|
|
hf-name: 'xlm-roberta-large-ViT-H-14__frozen_laion5b_s13b_b90k'
|
|
source: 'openclip'
|
|
runner: 'pokedex-medium'
|