Update ominicontrol.py

ominicontrol.py

@@ -12,7 +12,7 @@ print("Loading model...")
 pipe = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16
 )
-pipe = pipe.to("cuda")
+#pipe = pipe.to("cuda")
 
 pipe.unload_lora_weights()
 
@@ -37,6 +37,8 @@ pipe.load_lora_weights(
     adapter_name="snoopy",
 )
 
+pipe.enable_sequential_cpu_offload()
+
 
 def generate_image(
     image,