Update app.py

app.py

@@ -12,7 +12,7 @@ torch.backends.cuda.matmul.allow_tf32 = False
 # Initialize the base model and specific LoRA
 base_model = "black-forest-labs/FLUX.1-dev"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
-
+pipe.enable_model_cpu_offload()
 # pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
 
 pipe.load_lora_weights("Shakker-Labs/AWPortraitCN2", weight_name="AWPortraitCN_2.safetensors", low_cpu_mem_usage=True)