wooyeolbaek
commited on
Commit
•
5f4f504
1
Parent(s):
cf5a83b
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
|
|
17 |
)
|
18 |
pipe.unet = register_cross_attention_hook(pipe.unet)
|
19 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
20 |
-
pipe = pipe.to(
|
21 |
|
22 |
|
23 |
def inference(prompt):
|
|
|
17 |
)
|
18 |
pipe.unet = register_cross_attention_hook(pipe.unet)
|
19 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
20 |
+
pipe = pipe.to(device)
|
21 |
|
22 |
|
23 |
def inference(prompt):
|