Update infer.py

infer.py

@@ -39,7 +39,8 @@ print("Creating pipeline...")
 pipeline = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev", transformer=transformer, torch_dtype=torch.bfloat16
     , use_safetensors=True).to(DEVICE)
-pipeline.enable_sequential_cpu_offload()
+if USE_CPU_OFFLOAD:
+    pipeline.enable_sequential_cpu_offload()
 print("Generating image...")
 # Params:
 # prompt – The prompt or prompts to guide the image generation. If not defined, one has to pass prompt_embeds. instead.