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

Fixing conversation test for torch 1.8 (#10545)

Browse Source
This commit is contained in:
Nicolas Patry
2021-03-05 15:24:14 +01:00
committed by GitHub
parent dc9aaa3848
commit 54e55b52d4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_pipelines_conversational.py
View File

@@ -53,9 +53,9 @@ class SimpleConversationPipelineTests(unittest.TestCase):
model = GPT2LMHeadModel(config) model = GPT2LMHeadModel(config)
# Force model output to be L # Force model output to be L
V, D = model.lm_head.weight.shape V, D = model.lm_head.weight.shape
bias = torch.zeros(V, requires_grad=True) bias = torch.zeros(V)
weight = torch.zeros((V, D), requires_grad=True)
bias[76] = 1 bias[76] = 1
weight = torch.zeros((V, D), requires_grad=True)
model.lm_head.bias = torch.nn.Parameter(bias) model.lm_head.bias = torch.nn.Parameter(bias)
model.lm_head.weight = torch.nn.Parameter(weight) model.lm_head.weight = torch.nn.Parameter(weight)