From e391e252a11b724655626b9f824bff41d03e855d Mon Sep 17 00:00:00 2001 From: Zymrael Date: Wed, 10 Jan 2024 20:38:12 +0000 Subject: [PATCH] minor: remove list tensors --- src/engine.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/engine.py b/src/engine.py index 775bed2..557d7a1 100644 --- a/src/engine.py +++ b/src/engine.py @@ -292,7 +292,6 @@ def prefill_via_direct_recurrence( state = poles * state + x1v_[:, :, i] output[:, :, i] = torch.sum(residues * state, dim=-2)[..., 0] # .real - list_tensors(self.layer_idx) inference_params.state_dict[self.layer_idx] = torch.view_as_complex(state) return output