Spaces:

We-Want-GPU
/

diffusers-cross-attention-map-SDXL-t2i

Sleeping

wooyeolbaek commited on Mar 6, 2024

Commit

a40251e

verified ·

1 Parent(s): 5706abf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,8 +15,8 @@ from utils import (
 cross_attn_init()
 pipe = StableDiffusionXLPipeline.from_pretrained(
-    # "stabilityai/stable-diffusion-xl-base-1.0",
-    "stabilityai/sdxl-turbo",
     torch_dtype=torch.float16,
 )
 pipe.unet = register_cross_attention_hook(pipe.unet)

 cross_attn_init()
 pipe = StableDiffusionXLPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0",
+    # "stabilityai/sdxl-turbo",
     torch_dtype=torch.float16,
 )
 pipe.unet = register_cross_attention_hook(pipe.unet)