nick_93 commited on
Commit
b47e7c8
1 Parent(s): f678172
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ from diffusers import StableDiffusionXLPipeline
14
  import gradio as gr
15
  import gc
16
 
17
- device = "cpu"#"cuda"
18
- dtype = torch.float32#torch.float16
19
 
20
 
21
  css = """
 
14
  import gradio as gr
15
  import gc
16
 
17
+ device = "cuda"
18
+ dtype = torch.float16
19
 
20
 
21
  css = """