Blip: get/set input embeddings correctly (#34152)

* set-get embeds * add tests * fix tests * remove * return dict True * fix tests * why did i remove this * enabel torchscript tests
2024-11-01 08:39:39 +01:00
parent b53e44e847
commit 6beb3f1691
8 changed files with 288 additions and 32 deletions
--- a/tests/models/instructblipvideo/test_modeling_instructblipvideo.py
+++ b/tests/models/instructblipvideo/test_modeling_instructblipvideo.py
@@ -479,7 +479,7 @@ class InstructBlipVideoForConditionalGenerationDecoderOnlyTest(
    fx_compatible = False
    test_head_masking = False
    test_pruning = False
-    test_resize_embeddings = False
+    test_resize_embeddings = True
    test_attention_outputs = False
    test_torchscript = False
    _is_composite = True