ford442 commited on
Commit
8a1599f
·
verified ·
1 Parent(s): 32b1cbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 sageattnage==True:
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 sageattnage==True:
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 sageattnage==True:
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