Spaces:

We-Want-GPU
/

diffusers-cross-attention-map-SDXL-t2i

Runtime error

wooyeolbaek commited on Mar 6, 2024

Commit

1ef8448

verified ·

1 Parent(s): 1122931

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,8 @@ from utils import (
 cross_attn_init()
 pipe = StableDiffusionXLPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
     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)