Update app.py
Browse files
app.py
CHANGED
|
@@ -18,8 +18,7 @@ good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfold
|
|
| 18 |
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype, vae=taef1).to(device)
|
| 19 |
|
| 20 |
# Performance optimizations
|
| 21 |
-
|
| 22 |
-
pipe.enable_model_cpu_offload()
|
| 23 |
if hasattr(pipe, "enable_attention_slicing"):
|
| 24 |
pipe.enable_attention_slicing(1)
|
| 25 |
if hasattr(pipe, "enable_vae_slicing"):
|
|
|
|
| 18 |
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype, vae=taef1).to(device)
|
| 19 |
|
| 20 |
# Performance optimizations
|
| 21 |
+
|
|
|
|
| 22 |
if hasattr(pipe, "enable_attention_slicing"):
|
| 23 |
pipe.enable_attention_slicing(1)
|
| 24 |
if hasattr(pipe, "enable_vae_slicing"):
|