Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -407,7 +407,7 @@ def generate_30(
|
|
407 |
use_resolution_binning: bool = True,
|
408 |
progress=gr.Progress(track_tqdm=True)
|
409 |
):
|
410 |
-
if
|
411 |
F.scaled_dot_product_attention = sageattn
|
412 |
if sage==False:
|
413 |
F.scaled_dot_product_attention = F.scaled_dot_product_attention
|
@@ -461,7 +461,7 @@ def generate_60(
|
|
461 |
use_resolution_binning: bool = True,
|
462 |
progress=gr.Progress(track_tqdm=True)
|
463 |
):
|
464 |
-
if
|
465 |
F.scaled_dot_product_attention = sageattn
|
466 |
if sage==False:
|
467 |
F.scaled_dot_product_attention = F.scaled_dot_product_attention
|
@@ -510,7 +510,7 @@ def generate_90(
|
|
510 |
use_resolution_binning: bool = True,
|
511 |
progress=gr.Progress(track_tqdm=True)
|
512 |
):
|
513 |
-
if
|
514 |
F.scaled_dot_product_attention = sageattn
|
515 |
if sage==False:
|
516 |
F.scaled_dot_product_attention = F.scaled_dot_product_attention
|
|
|
407 |
use_resolution_binning: bool = True,
|
408 |
progress=gr.Progress(track_tqdm=True)
|
409 |
):
|
410 |
+
if sage==True:
|
411 |
F.scaled_dot_product_attention = sageattn
|
412 |
if sage==False:
|
413 |
F.scaled_dot_product_attention = F.scaled_dot_product_attention
|
|
|
461 |
use_resolution_binning: bool = True,
|
462 |
progress=gr.Progress(track_tqdm=True)
|
463 |
):
|
464 |
+
if sage==True:
|
465 |
F.scaled_dot_product_attention = sageattn
|
466 |
if sage==False:
|
467 |
F.scaled_dot_product_attention = F.scaled_dot_product_attention
|
|
|
510 |
use_resolution_binning: bool = True,
|
511 |
progress=gr.Progress(track_tqdm=True)
|
512 |
):
|
513 |
+
if sage==True:
|
514 |
F.scaled_dot_product_attention = sageattn
|
515 |
if sage==False:
|
516 |
F.scaled_dot_product_attention = F.scaled_dot_product_attention
|