Spaces:
Sleeping
Sleeping
File size: 682 Bytes
c4ef1cf 9513cca c4ef1cf 9513cca c4ef1cf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
"""
Embedding module - Visual and text embedding generation.
Provides:
- VisualEmbedder: Backend-agnostic visual embedder (ColPali, etc.)
- Pooling utilities: tile-level, global, MaxSim scoring
"""
from visual_rag.embedding.pooling import (
compute_maxsim_batch,
compute_maxsim_score,
global_mean_pooling,
tile_level_mean_pooling,
)
from visual_rag.embedding.visual_embedder import ColPaliEmbedder, VisualEmbedder
__all__ = [
# Main embedder
"VisualEmbedder",
"ColPaliEmbedder", # Backward compatibility alias
# Pooling functions
"tile_level_mean_pooling",
"global_mean_pooling",
"compute_maxsim_score",
"compute_maxsim_batch",
]
|