Rahul8827 commited on
Commit
b329693
1 Parent(s): 0af4597

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from diffusers import StableDiffusionPipeline
4
 
5
 
6
  class CFG:
7
- device = "cuda"
8
  seed = 42
9
  generator = torch.Generator(device).manual_seed(seed)
10
  image_gen_steps = 35
 
4
 
5
 
6
  class CFG:
7
+ device = "cpu"
8
  seed = 42
9
  generator = torch.Generator(device).manual_seed(seed)
10
  image_gen_steps = 35