KeyError: 'up_cross'

[tanlingp]

Sorry to interrupt, code error Traceback (most recent call last). File "main.py", line 153, in adv_image, clean_acc, adv_acc = run_diffusion_attack(tmp_image, label[ind:ind + 1]. File "main.py", line 72, in run_diffusion_attack adv_image, clean_acc, adv_acc = diff_latent_attack.diffattack(diffusion_model, label, controller. File "/usr/local/lib/python3.8/dist-packages/torch/autograd/grad_mode.py", line 27, in decorate_context return func(*args, **kwargs) File "/202221623173/DiffAttack-main/diff_latent_attack.py", line 369, in diffattack before_attention_map = aggregate_attention(prompt, controller, 7, ("up", "down"), True, 0, is_cpu=False) File "/202221623173/DiffAttack-main/utils.py", line 13, in aggregate_attention for item in attention_maps[f"{location}_{'cross' if is_cross else 'self'}"]. KeyError: 'up_cross'