Skip to content

Passage version 1.2

Victor Amblard requested to merge version-1.2 into prod
  • [FIX] OOM when using the method embed_images with a large number of images => using batching now
  • [FIX] Correction d'un bug lié à l'absence d'argument device passé dans l'instanciation d'une classe Embedder dans la méthode load_embeddings
  • [IMPROVEMENT] Remove dependencies to sentence_transformers and clip libraries to rely solely on transformers lib => renaming CLIPEmbedder -> CLIPB32Embedder and removal of CLIPHFEmbedder
  • [IMPROVEMENT] Improved typing and added decorators @abstracmethod when needed + declared Embedder class as an abstract class
  • [IMPROVEMENT] Using logging (structlog) instead of prints
  • [REFACTORING] Created a parent class CLIPEmbedder(Embedder) from which CLIPL14Embedding and CLIPB32Embedding inherit to reduce duplicated code
Edited by Victor Amblard

Merge request reports

Loading