Update app.py

app.py

@@ -25,19 +25,8 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 
-if torch.cuda.is_available():
-    pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium", torch_dtype=torch.float16, use_auth_token=huggingface_token)
+pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium", torch_dtype=torch.float16, use_auth_token=huggingface_token)
     
-    if ENABLE_CPU_OFFLOAD:
-        pipe.enable_model_cpu_offload()
-    else:
-        pipe.to(device)    
-        print("Loaded on Device!")
-    
-    if USE_TORCH_COMPILE:
-        pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
-        print("Model Compiled!")
-
 
 def save_image(img):
     unique_name = str(uuid.uuid4()) + ".png"
@@ -84,7 +73,6 @@ def generate(
         num_inference_steps=num_inference_steps,
         generator=generator,
         num_images_per_prompt=NUM_IMAGES_PER_PROMPT,
-        use_resolution_binning=use_resolution_binning,
         output_type="pil",
     ).images