10 Commits

Author	SHA1	Message	Date
Madeesh Kannan	7376838922	feat!: Framework-agnostic device management (#6748) ... * feat: Framework-agnostic device management * Add release note * Linting * Fix test * Add `first_device` property, expand release notes, validate `ComponentDevice` state	2024-01-17 10:41:34 +01:00
ZanSara	0616197b44	feat!: Rename model_name_or_path to model in TransformersSimilarityRanker (#6734) ... * rename model parameter in transformers ranker * fix tests for transformers ranker * reno * reno * typo	2024-01-12 17:09:12 +01:00
Stefano Fiorucci	80c3e6825a	fix: serialize/deserialize torch dtype in the components that need it (#6713) ... * first draft for ranker * same for the reader * consider also bnb_4bit_compute_dtype * dtype serialization in hugging_face_local_generator * add release note * address dtype defined in huggingface_pipeline_kwargs * test quantization options in reader * fix * serialize quantization_config * test quantization_config serialization * address feedback * fix typo	2024-01-12 12:22:45 +01:00
Sebastian Husch Lee	beade1cef9	feat: Add scaling and thresholding of the similarity ranker scores (#6683) ... * Add scale_score functionality to the TransformersSimilarityRanker * Updated test to check scores * Use pytest approx when comparing floats * Updated how scale score works and added calibration factor. Started to add score threshold. * Add support for score_threshold * Add some parameters to the run method * Add release notes * Fix mypy * Be more tolerant on the score values * Adding unit test for scale_score=False * Add unit test for score threshold * Update tests * Rename test * Fix typo * PR comments	2024-01-08 09:05:24 +01:00
Sebastian Husch Lee	c294b8ac8c	feat: Add auto device checks and model_kwargs to TransformersSimilarityRanker (#6561) ... * Add device checking and model_kwargs like we do in ExtractiveReader * Add release notes * Make a utility function for the device checking * Better warning message and updated ExtractiveReader to use the util function * Add unit tests for get_device * Fix pylint	2023-12-18 15:13:42 +01:00
Sebastian Husch Lee	3e0e81b1e0	feat: Add meta_fields_to_embed to TransformersSimilarityRanker (#6564) ... * Add initial implementation following SentenceTransformersDocumentEmbedder * Add test for embedding metadata * Add release notes * Update name * Fix tests and to dict * Fix release notes	2023-12-18 11:28:16 +01:00
bogdankostic	728383a149	fix: Make TransformersSimilarityRanker run with single document list (#6503) ... * Make `TransformersSimilarityRanker` run with single document list * Add release note * Remove unused import in test	2023-12-08 16:18:46 +01:00
Massimiliano Pippi	7c05f37a53	remove unit marker (#6450)	2023-11-29 19:24:25 +01:00
Silvano Cerza	e6637f5ec2	Fix all tests	2023-11-24 14:48:43 +01:00
Massimiliano Pippi	8adb8bbab8	Remove preview folder in test/ ... --------- Co-authored-by: Silvano Cerza <silvanocerza@gmail.com>	2023-11-24 11:52:55 +01:00