This commit is contained in:
cocktailpeanut
2024-08-08 20:04:19 -04:00
parent b63d42a4e2
commit fd862ac7f8

4
app.py
View File

@@ -59,8 +59,8 @@ def infer(prompt, checkpoint="black-forest-labs/FLUX.1-schnell", seed=42, guidan
if device == "cuda":
print(f"enable model cpu offload...")
pipe.enable_model_cpu_offload()
#pipe.enable_sequential_cpu_offload()
#pipe.enable_model_cpu_offload()
pipe.enable_sequential_cpu_offload()
print(f"done!")
selected = checkpoint
if randomize_seed: