SuCicada commited on
Commit
3a3cc9a
2 Parent(s): 831a077 7a69c1b

Merge remote-tracking branch 'origin/main'

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -7,6 +7,8 @@ from torch import autocast
7
  from diffusers.pipelines.stable_diffusion import StableDiffusionPipeline
8
 
9
  model_id = "hakurei/waifu-diffusion"
 
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
 
12
 
 
7
  from diffusers.pipelines.stable_diffusion import StableDiffusionPipeline
8
 
9
  model_id = "hakurei/waifu-diffusion"
10
+ torch.backends.cudnn.benchmark = True
11
+
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
 
14