X commited on
Commit
75dd650
·
verified ·
1 Parent(s): 0cfac41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
16
  safety_checker=None,
17
  requires_safety_checker=False
18
  )
19
- pipe = pipe.to("cuda")
20
  pipe.enable_attention_slicing()
21
  pipe.enable_vae_slicing()
22
  print("Модель загружена!")
 
16
  safety_checker=None,
17
  requires_safety_checker=False
18
  )
19
+ pipe = pipe.to("cpu")
20
  pipe.enable_attention_slicing()
21
  pipe.enable_vae_slicing()
22
  print("Модель загружена!")