Skip to content

Commit 6788aa4

Browse files
committed
reverted backend=sendnn to backend=sendnn_decoder
Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
1 parent b2fc820 commit 6788aa4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/test_decoders.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def test_common_shapes(model_path, batch_size, seq_length, max_new_tokens):
141141

142142
model.eval()
143143
torch.set_grad_enabled(False)
144-
model.compile(backend="sendnn")
144+
model.compile(backend="sendnn_decoder")
145145

146146
# prepare the cpu model
147147
validation_model = get_model(

0 commit comments

Comments
 (0)