Rahul8827 commited on
Commit
abc4840
·
1 Parent(s): 023bdfe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -8,6 +8,8 @@ from glide_text2im.model_creation import (
8
  model_and_diffusion_defaults,
9
  model_and_diffusion_defaults_upsampler
10
  )
 
 
11
  # Create base model.
12
  options = model_and_diffusion_defaults()
13
  options['use_fp16'] = has_cuda
 
8
  model_and_diffusion_defaults,
9
  model_and_diffusion_defaults_upsampler
10
  )
11
+ has_cuda = th.cuda.is_available()
12
+ device = th.device('cpu' if not has_cuda else 'cuda')
13
  # Create base model.
14
  options = model_and_diffusion_defaults()
15
  options['use_fp16'] = has_cuda