down_cross

Getting this error:

wapping_class.py in call(self, x_t, attention_store)
24 if self.counter > self.start_blend:
25
---> 26 maps = attention_store["down_cross"][2:4] + attention_store["up_cross"][:3]
27 maps = [item.reshape(self.alpha_layers.shape[0], -1, 1, 16, 16, 77) for item in maps]
28 maps = torch.cat(maps, dim=1)

KeyError: 'down_cross'