nateraw commited on
Commit
5b675db
1 Parent(s): b7321c6

use less vram

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -15,6 +15,8 @@ pipeline = StableDiffusionWalkPipeline.from_pretrained(
15
  beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear"
16
  )
17
  ).to("cuda")
 
 
18
  interface = Interface(pipeline)
19
 
20
  if __name__ == '__main__':
 
15
  beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear"
16
  )
17
  ).to("cuda")
18
+ pipeline.enable_attention_slicing()
19
+
20
  interface = Interface(pipeline)
21
 
22
  if __name__ == '__main__':