riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix DeBERTa + Conversational pipeline slow tests (#10743)

Browse Source 
* Fix DeBERTa-v2 variable assignment

* Fix conversational pipeline test
This commit is contained in:
Lysandre Debut 2021-03-16 11:18:20 -04:00 committed by GitHub
parent d3d388b934
commit 1449222217
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/deberta_v2/modeling_deberta_v2.py
View File

@ -573,7 +573,7 @@ class DisentangledSelfAttention(torch.nn.Module):
self.value_proj = nn.Linear(config.hidden_size, self.all_head_size, bias=True)
self.share_att_key = getattr(config, "share_att_key", False)
self.pos_att_type = config.pos_att_type
self.pos_att_type = config.pos_att_type if config.pos_att_type is not None else []
self.relative_attention = getattr(config, "relative_attention", False)
if self.relative_attention:

2
src/transformers/pipelines/conversational.py
View File

@ -340,6 +340,6 @@ class ConversationalPipeline(Pipeline):
# If the tokenizer cannot handle conversations, we default to only the old version
input_ids = [self._legacy_parse_and_tokenize(conversation) for conversation in conversations]
inputs = self.tokenizer.pad(
{"input_ids": input_ids}, padding="longest", return_attention_mask=True, return_tensors="pt"
{"input_ids": input_ids}, padding="longest", return_attention_mask=True, return_tensors=self.framework
)
return inputs