Spaces:
Paused
Paused
Update app.py
Browse files
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 |
-
|
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()
|