Spaces:

tight-inversion
/

tight-inversion-pulid-demo

Running on Zero

tight-inversion commited on Mar 6

Commit

56a3023

1 Parent(s): 1a8a74e

Fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -194,7 +194,7 @@ def generate_image(
             neg_txt=inp_neg["txt"] if use_true_cfg else None,
             neg_txt_ids=inp_neg["txt_ids"] if use_true_cfg else None,
             neg_vec=inp_neg["vec"] if use_true_cfg else None,
-            aggressive_offload=flux_generator.aggressive_offload,
             y_1=noise,
             gamma=gamma
         )
@@ -221,7 +221,7 @@ def generate_image(
             neg_txt=inp_neg["txt"] if use_true_cfg else None,
             neg_txt_ids=inp_neg["txt_ids"] if use_true_cfg else None,
             neg_vec=inp_neg["vec"] if use_true_cfg else None,
-            aggressive_offload=flux_generator.aggressive_offload,
             y_0=y_0,
             eta=eta,
             s=s,
@@ -244,7 +244,7 @@ def generate_image(
             neg_txt=inp_neg["txt"] if use_true_cfg else None,
             neg_txt_ids=inp_neg["txt_ids"] if use_true_cfg else None,
             neg_vec=inp_neg["vec"] if use_true_cfg else None,
-            aggressive_offload=flux_generator.aggressive_offload,
             y_0=y_0,
             eta=eta,
             s=s,

             neg_txt=inp_neg["txt"] if use_true_cfg else None,
             neg_txt_ids=inp_neg["txt_ids"] if use_true_cfg else None,
             neg_vec=inp_neg["vec"] if use_true_cfg else None,
+            aggressive_offload=False,
             y_1=noise,
             gamma=gamma
         )
             neg_txt=inp_neg["txt"] if use_true_cfg else None,
             neg_txt_ids=inp_neg["txt_ids"] if use_true_cfg else None,
             neg_vec=inp_neg["vec"] if use_true_cfg else None,
+            aggressive_offload=False,
             y_0=y_0,
             eta=eta,
             s=s,
             neg_txt=inp_neg["txt"] if use_true_cfg else None,
             neg_txt_ids=inp_neg["txt_ids"] if use_true_cfg else None,
             neg_vec=inp_neg["vec"] if use_true_cfg else None,
+            aggressive_offload=False,
             y_0=y_0,
             eta=eta,
             s=s,