transformers/tests/models/grounding_dino
Eduardo Pacheco c651ea982b
[Grounding DINO] Add support for cross-attention in GroundingDinoMultiHeadAttention (#30364)
* Added cross attention support

* Fixed dtypes

* Fixed assumption

* Moved to decoder
2024-04-23 09:56:14 +01:00
..
__init__.py Adding grounding dino (#26087) 2024-04-11 08:32:16 +01:00
test_image_processing_grounding_dino.py Adding grounding dino (#26087) 2024-04-11 08:32:16 +01:00
test_modeling_grounding_dino.py [Grounding DINO] Add support for cross-attention in GroundingDinoMultiHeadAttention (#30364) 2024-04-23 09:56:14 +01:00
test_processor_grounding_dino.py Adding grounding dino (#26087) 2024-04-11 08:32:16 +01:00