update

app.py

@@ -75,7 +75,8 @@ def init():
     ).to(DEVICE)
     pipe.text_encoder_2 = text_encoder_2
     pipe.transformer = transformer
-    pipe.enable_model_cpu_offload()
+    if DEVICE == "cuda":
+        pipe.enable_model_cpu_offload()
 
 #    generator = torch.Generator().manual_seed(12345)
 #    image = pipe(
@@ -194,4 +195,5 @@ with gr.Blocks(css=css) as demo:
         outputs = [result, seed]
     )
 
+init()
 demo.launch()

requirements.txt

@@ -1,7 +1,7 @@
 accelerate
 #git+https://github.com/huggingface/diffusers.git@flux-pipeline
-#git+https://github.com/peanutcocktail/diffusers.git
-git+https://github.com/huggingface/diffusers.git
+git+https://github.com/peanutcocktail/diffusers.git
+#git+https://github.com/huggingface/diffusers.git
 invisible_watermark
 #torch
 transformers==4.42.4