Update app.py

app.py

@@ -25,23 +25,13 @@ ENHANCE_STEPS = 2 # Fixed steps for the enhance button
 dtype = torch.float16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = FluxWithCFGPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype
-)
+pipe = FluxWithCFGPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype)
 pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype)
 
-pipe.to(device)
-
-# Apply optimizations
-pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v2.3.safetensors', adapter_name="better")
-pipe.set_adapters(["better"], adapter_weights=[1.0])
-pipe.fuse_lora(adapter_name=["better"], lora_scale=1.0) # Fuse for potential speedup
-pipe.unload_lora_weights() # Unload after fusing
-
-# --- Compilation (Major Speed Optimization) ---
-# pipe.vae.decoder = torch.compile(pipe.vae.decoder, mode="reduce-overhead", fullgraph=True)
-# pipe.vae.encoder = torch.compile(pipe.vae.encoder, mode="reduce-overhead", fullgraph=True)
+pipe.vae.enable_slicing()
+pipe.vae.enable_tiling()
 
+pipe.to(device)
 
 # --- Inference Function ---
 @spaces.GPU