Update app.py

app.py

@@ -12,8 +12,8 @@ import random
 import re
 import sys
 import warnings
-ENABLE_COMPILE = os.environ.get("ENABLE_COMPILE", "true").lower() == "true"
-ENABLE_WARMUP = os.environ.get("ENABLE_WARMUP", "true").lower() == "true"
+# ENABLE_COMPILE = os.environ.get("ENABLE_COMPILE", "true").lower() == "true"
+# ENABLE_WARMUP = os.environ.get("ENABLE_WARMUP", "true").lower() == "true"
 
 # Load the pipeline once at startup
 print("Loading Z-Image-Turbo pipeline...")
@@ -27,9 +27,9 @@ pipe = DiffusionPipeline.from_pretrained(
 
 #pipe.transformer.set_attention_backend(attention_backend)
 
-if enable_compile:
-    print("Compiling transformer...")
-    pipe.transformer = torch.compile(pipe.transformer, mode="max-autotune-no-cudagraphs", fullgraph=False)
+# if enable_compile:
+#     print("Compiling transformer...")
+#     pipe.transformer = torch.compile(pipe.transformer, mode="max-autotune-no-cudagraphs", fullgraph=False)
 
 pipe.to("cuda")