SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
* set-get embeds

* add tests

* fix tests

* remove

* return dict True

* fix tests

* why did i remove this

* enabel torchscript tests
This commit is contained in:
Raushan Turganbay
2024-11-01 08:39:39 +01:00
committed by GitHub
parent b53e44e847
commit 6beb3f1691
8 changed files with 288 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_common.py
View File

@@ -1811,6 +1811,7 @@ class ModelTesterMixin:
original_config,
inputs_dict,
) = self.model_tester.prepare_config_and_inputs_for_common()
inputs_dict.pop("labels", None)
for model_class in self.all_model_classes:
config = copy.deepcopy(original_config)
@@ -1988,6 +1989,7 @@ class ModelTesterMixin:
original_config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
original_config.tie_word_embeddings = False
inputs_dict.pop("labels", None)
# if model cannot untied embeddings -> leave test
if original_config.tie_word_embeddings: