瀏覽代碼

unused lines

justheuristic 3 年之前
父節點
當前提交
1f5afd1cdf
共有 1 個文件被更改,包括 0 次插入2 次删除
  1. 0 2
      tests/test_remote_sequential.py

+ 0 - 2
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)