Update app.py

app.py

@@ -19,9 +19,7 @@ if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>⚠️ This space is running on the CPU. This demo doesn't work on CPU 😞! Run on a GPU by duplicating this space or test our website for free and unlimited by <a href='https://squaadai.com'>clicking here</a>, which provides these and more options.</p>"
 
 MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1824"))
-USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "0"
-ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
+MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1824")
 ENABLE_USE_LORA = os.getenv("ENABLE_USE_LORA", "1") == "1"
 ENABLE_USE_VAE = os.getenv("ENABLE_USE_VAE", "1") == "1"
 ENABLE_USE_IMG2IMG = os.getenv("ENABLE_USE_VAE", "1") == "1"
@@ -87,9 +85,6 @@ def generate(
             
         else:
             pipe.to(device)
-
-        if USE_TORCH_COMPILE:
-            pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
         
     generator = torch.Generator().manual_seed(seed)