Update app.py

app.py

@@ -20,7 +20,8 @@ lora_base_path = "./models"
 pipe = FluxPipeline.from_pretrained(base_path, torch_dtype=torch.bfloat16)
 transformer = FluxTransformer2DModel.from_pretrained(base_path, subfolder="transformer", torch_dtype=torch.bfloat16)
 pipe.transformer = transformer
-pipe.to("cuda")
+pipe.enable_sequential_cpu_offload()
+#pipe.to("cuda")
 
 def clear_cache(transformer):
     for name, attn_processor in transformer.attn_processors.items():