Spaces:
Sleeping
Sleeping
genevera
commited on
Commit
•
0c0d5ca
1
Parent(s):
1a8a5f1
dont use xformers if cuda isnt available
Browse files
app.py
CHANGED
@@ -228,7 +228,9 @@ def greet(audio, steps=25, scheduler="ddpm"):
|
|
228 |
scheduler=use_sched,
|
229 |
safety_checker=None,
|
230 |
).to(device)
|
231 |
-
pipeline.
|
|
|
|
|
232 |
|
233 |
# print(f"taking {steps} steps using the {scheduler} scheduler")
|
234 |
image = pipeline(
|
|
|
228 |
scheduler=use_sched,
|
229 |
safety_checker=None,
|
230 |
).to(device)
|
231 |
+
pipeline.enable_attention_slicing()
|
232 |
+
if torch.cuda.is_available():
|
233 |
+
pipeline.enable_xformers_memory_efficient_attention()
|
234 |
|
235 |
# print(f"taking {steps} steps using the {scheduler} scheduler")
|
236 |
image = pipeline(
|