Raushan Turganbay 9725e5be2f Pixtral: vectorize patch embeddings and enable tests (#35122) ... * initial POC * - batch mix feature * fix tests * fix tests * make style * do not skip and instead fix tests * update * return back the test * correct text with the correct ckpt		2025-01-30 12:40:18 +01:00
..
agents	use torch.testing.assertclose instead to get more details about error in cis (#35659)	2025-01-24 16:55:28 +01:00
bettertransformer	use torch.testing.assertclose instead to get more details about error in cis (#35659)	2025-01-24 16:55:28 +01:00
deepspeed	use torch.testing.assertclose instead to get more details about error in cis (#35659)	2025-01-24 16:55:28 +01:00
extended	[tests] skip tests for xpu (#33553)	2024-09-19 19:28:04 +01:00
fixtures	Implementation of SuperPoint and AutoModelForKeypointDetection (#28966)	2024-03-19 14:43:02 +00:00
fsdp	[tests] make cuda-only tests device-agnostic (#35607)	2025-01-13 14:48:39 +01:00
generation	[generate] move max time tests (#35962)	2025-01-29 17:56:46 +00:00
models	Pixtral: vectorize patch embeddings and enable tests (#35122)	2025-01-30 12:40:18 +01:00
optimization	Update unwrap_and_save_reload_schedule to use weights_only=False (#35952)	2025-01-29 14:30:57 +01:00
peft_integration	use torch.testing.assertclose instead to get more details about error in cis (#35659)	2025-01-24 16:55:28 +01:00
pipelines	Output dicts support in text generation pipeline (#35092)	2025-01-29 14:44:46 +00:00
quantization	Split and clean up GGUF quantization tests (#35502)	2025-01-27 15:46:57 +01:00
repo_utils	Fix modular edge case + modular sorting order (#35562)	2025-01-09 17:17:52 +01:00
sagemaker	Trainer - deprecate tokenizer for processing_class (#32385)	2024-10-02 14:08:46 +01:00
tokenization	tokenizer train from iterator without pre_tokenizers (#35396)	2025-01-09 15:34:43 +01:00
tp	Fix TP initialization (#35860)	2025-01-28 15:07:37 +01:00
trainer	use torch.testing.assertclose instead to get more details about error in cis (#35659)	2025-01-24 16:55:28 +01:00
utils	Test: generate with torch.compile(model.forward) as a fast test (#34544)	2025-01-28 14:10:38 +00:00
__init__.py
test_backbone_common.py	Align backbone stage selection with out_indices & out_features (#27606)	2023-12-20 18:33:17 +00:00
test_configuration_common.py	Load sub-configs from composite configs (#34410)	2024-11-05 11:34:01 +01:00
test_feature_extraction_common.py
test_image_processing_common.py	use torch.testing.assertclose instead to get more details about error in cis (#35659)	2025-01-24 16:55:28 +01:00
test_image_transforms.py	fix: center_crop occasionally outputs off-by-one dimension matrix (#30934)	2024-05-21 13:56:52 +01:00
test_modeling_common.py	Pixtral: vectorize patch embeddings and enable tests (#35122)	2025-01-30 12:40:18 +01:00
test_modeling_flax_common.py	🚨All attention refactor🚨 (#35235)	2024-12-18 16:53:39 +01:00
test_modeling_tf_common.py	🚨All attention refactor🚨 (#35235)	2024-12-18 16:53:39 +01:00
test_pipeline_mixin.py	Add image text to text pipeline (#34170)	2024-10-31 15:48:11 -04:00
test_processing_common.py	VLMs: major clean up 🧼 (#34502)	2025-01-08 10:35:23 +01:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py	[tokenizers] Ensure that add_prefix_space is propagated to backend_tokenizer.pre_tokenizer (#35593)	2025-01-09 17:46:50 +01:00