Update app.py

app.py

@@ -407,9 +407,9 @@ def generate_30(
     use_resolution_binning: bool = True, 
     progress=gr.Progress(track_tqdm=True)
 ):
-    if Sage==True:
+    if sageattnage==True:
         F.scaled_dot_product_attention = sageattn
-    if Sage==False:
+    if sage==False:
         F.scaled_dot_product_attention = F.scaled_dot_product_attention
     seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device='cuda').manual_seed(seed)
@@ -461,6 +461,10 @@ def generate_60(
     use_resolution_binning: bool = True, 
     progress=gr.Progress(track_tqdm=True)
 ):
+    if sageattnage==True:
+        F.scaled_dot_product_attention = sageattn
+    if sage==False:
+        F.scaled_dot_product_attention = F.scaled_dot_product_attention
     seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device='cuda').manual_seed(seed)
     options = {
@@ -506,6 +510,10 @@ def generate_90(
     use_resolution_binning: bool = True, 
     progress=gr.Progress(track_tqdm=True)
 ):
+    if sageattnage==True:
+        F.scaled_dot_product_attention = sageattn
+    if sage==False:
+        F.scaled_dot_product_attention = F.scaled_dot_product_attention
     seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device='cuda').manual_seed(seed)
     options = {