Spaces:

We-Want-GPU
/

diffusers-cross-attention-map-SDXL-t2i

Sleeping

wooyeolbaek commited on Nov 28, 2024

Commit

602d244

•

1 Parent(s): c993844

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,11 +26,11 @@ pipe = pipe.to(device)
 def inference(prompt):
     image = pipe(
-        prompt,
         num_inference_steps=15,
     ).images[0]
-    total_attn_maps = save_attention_maps(attn_maps, pipe.tokenizer, prompt)
     return image, total_attn_maps

 def inference(prompt):
     image = pipe(
+        [prompt],
         num_inference_steps=15,
     ).images[0]
+    total_attn_maps = save_attention_maps(attn_maps, pipe.tokenizer, [prompt])
     return image, total_attn_maps