diff --git a/machine-learning/export/immich_model_exporter/exporters/constants.py b/machine-learning/export/immich_model_exporter/exporters/constants.py index 8871535ea1..ca72872858 100644 --- a/machine-learning/export/immich_model_exporter/exporters/constants.py +++ b/machine-learning/export/immich_model_exporter/exporters/constants.py @@ -24,18 +24,6 @@ SOURCE_TO_METADATA = { RKNN_SOCS = ["rk3566", "rk3576", "rk3588"] -# the builder hangs when using flash attention with these models -RKNN_VISUAL_FLASH_ATTENTION_BLACKLIST = { - "ViT-H-14-378-quickgelu__dfn5b", - "ViT-L-16-SigLIP-384__webli", - "ViT-L-16-SigLIP2-384__webli", - "ViT-L-16-SigLIP2-512__webli", - "ViT-SO400M-14-SigLIP-384__webli", - "ViT-SO400M-14-SigLIP2-378__webli", - "ViT-SO400M-16-SigLIP2-378__webli", - "ViT-SO400M-16-SigLIP2-512__webli", -} - # glob to delete old UUID blobs when reuploading models _uuid_char = "[a-fA-F0-9]" diff --git a/machine-learning/export/immich_model_exporter/exporters/rknn.py b/machine-learning/export/immich_model_exporter/exporters/rknn.py index e8a61c53ee..76f7a09365 100644 --- a/machine-learning/export/immich_model_exporter/exporters/rknn.py +++ b/machine-learning/export/immich_model_exporter/exporters/rknn.py @@ -22,7 +22,6 @@ def _export_platform( rknn = RKNN(verbose=False) - # flash_attention = model_dir.name != "visual" or model_dir.parent.name not in RKNN_VISUAL_FLASH_ATTENTION_BLACKLIST rknn.config( target_platform=target_platform, dynamic_input=dynamic_input,