wooyeolbaek commited on
Commit
5f4f504
1 Parent(s): cf5a83b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(dtype=torch.float32) if device == 'cpu' else pipe.to(device)
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):