update

2026-05-15 00:46:37 +08:00
parent b4c7697e8a
commit f64e624a39
8 changed files with 190 additions and 100 deletions
@@ -63,13 +63,14 @@ def make_flux_fill_predictor(

    if device == "cuda":
        try:
-            pipe.enable_model_cpu_offload()
+            pipe.enable_vae_tiling()
        except Exception:
-            try:
-                pipe.enable_vae_tiling()
-            except Exception:
-                pass
-            pipe.to("cuda")
+            pass
+        try:
+            pipe.enable_attention_slicing()
+        except Exception:
+            pass
+        pipe.to("cuda")
    else:
        pipe.to(device)