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

[Generate] Remove attention_mask and integrate model_main_input_name (#14856)

Browse Source 
* up

* save

* correct

* up

* correct more

* up

* up

* up

* up

* up

* correct

* fix tf

* fix

* remove tokenizer
This commit is contained in:
Patrick von Platen
2021-12-23 19:43:37 +01:00
committed by GitHub
parent 86b40073e9
commit fe4197ab11
8 changed files with 117 additions and 155 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_generation_utils.py
View File

@@ -1856,7 +1856,7 @@ class GenerationIntegrationTests(unittest.TestCase):
model = GPT2LMHeadModel.from_pretrained("hf-internal-testing/tiny-random-gpt2", max_length=10).to(torch_device)
input_ids = tokenizer(article, return_tensors="pt").input_ids.to(torch_device)
with self.assertRaises(ValueError):
model.generate(input_ids=input_ids, input_values=input_ids)
model.generate(input_ids=input_ids, inputs_embeds=input_ids)
def test_generate_input_values_as_encoder_kwarg(self):
input_values = floats_tensor((2, 250))