DonImages commited on
Commit
11ad404
·
verified ·
1 Parent(s): fa0a16f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import random
6
  from diffusers import StableDiffusion3Pipeline
7
  from diffusers.loaders import SD3LoraLoaderMixin
8
 
9
- # Device selection
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
12
 
 
6
  from diffusers import StableDiffusion3Pipeline
7
  from diffusers.loaders import SD3LoraLoaderMixin
8
 
9
+ # Device selection NOTE
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
12