[Add] Adaptive clustering

2025-07-10 20:35:46 +00:00
parent 65adb5d4ba
commit 8e1d0568a2
3 changed files with 137 additions and 85 deletions
--- a/src/embedding_extraction.py
+++ b/src/embedding_extraction.py
@@ -57,7 +57,7 @@ def get_layoutlm_image_embeddings(

    return embeddings.cpu().numpy(), valid_paths

-def get_image_embeddings(
+def get_qwenvl_image_embeddings(
    model: AutoModel,
    processor: AutoProcessor,
    image_paths: List[str],