Skip to content

GitLab

"docs/source/en/model_doc/starcoder2.md" did not exist on "c236a621729edb92dbb3e1dedd448be76cb82211"
Unverified Commit 54e55b52 authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Fixing conversation test for torch 1.8 (#10545)

parent dc9aaa38
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/test_pipelines_conversational.py
View file @ 54e55b52
...@@ -53,9 +53,9 @@ class SimpleConversationPipelineTests(unittest.TestCase): ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment