Noename commited on
Commit
c89bb38
·
verified ·
1 Parent(s): 4621db3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,10 +70,10 @@ pipe = pipe.to(device)
70
  # Apply optimizations based on hardware
71
  if device == "cuda":
72
  pipe = pipe.to(device)
73
- if enable_xformers:
74
  # pipe.enable_xformers_memory_efficient_attention()
75
  # print("xformers optimization enabled")
76
- pipe.enable_attention_slicing()
77
  elif device == "mps":
78
  pipe = pipe.to(device)
79
  pipe.enable_attention_slicing()
 
70
  # Apply optimizations based on hardware
71
  if device == "cuda":
72
  pipe = pipe.to(device)
73
+ # if enable_xformers:
74
  # pipe.enable_xformers_memory_efficient_attention()
75
  # print("xformers optimization enabled")
76
+ pipe.enable_attention_slicing()
77
  elif device == "mps":
78
  pipe = pipe.to(device)
79
  pipe.enable_attention_slicing()