fix hoook bug

app.py

@@ -111,7 +111,7 @@ def main():
 
         # get token maps from plain text to image generation.
         begin_time = time.time()
-        if model.attention_maps is None:
+        if model.selfattn_maps is None and model.crossattn_maps is None:
             model.register_tokenmap_hooks()
         else:
             model.reset_attention_maps()