minor: remove list tensors

togethercomputer · Jan 10, 2024 · e391e25 · e391e25
1 parent 75efed0
commit e391e25
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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