From 1f5afd1cdfc1b402110b1854c467cc97535f2f1b Mon Sep 17 00:00:00 2001 From: justheuristic Date: Fri, 12 Aug 2022 18:15:54 +0300 Subject: [PATCH] unused lines --- tests/test_remote_sequential.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/test_remote_sequential.py b/tests/test_remote_sequential.py index 11068c4..678ec01 100644 --- a/tests/test_remote_sequential.py +++ b/tests/test_remote_sequential.py @@ -57,8 +57,6 @@ def test_remote_sequential_prompts(batch_size=2, seq_len=5, pre_seq_len=3): input_prompts = input_prompts.detach().requires_grad_(True) intermediate_prompts = intermediate_prompts.detach().requires_grad_(True) - with torch.no_grad(): - intermediate_prompts[...] = torch.randn_like(intermediate_prompts) inputs_with_prompts = torch.cat([inputs, input_prompts], dim=1) assert inputs_with_prompts.shape == (batch_size, seq_len + pre_seq_len, config.hidden_size)