Update app.py

app.py

@@ -21,11 +21,6 @@ except ImportError:
 from PIL import Image
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 from diffusers import AutoencoderKL, UNet2DConditionModel, DDIMScheduler, EulerAncestralDiscreteScheduler, DPMSolverMultistepScheduler
-try:
-    from diffusers.models.attention_processor import AttnProcessor2_0
-except ImportError:
-    # Fallback for older diffusers versions
-    AttnProcessor2_0 = None
 from transformers import CLIPTextModel, CLIPTokenizer
 from enum import Enum
 import torch.nn as nn
@@ -137,9 +132,7 @@ if not os.path.exists(model_path):
         from huggingface_hub import hf_hub_download
         model_path = hf_hub_download(
             repo_id="lllyasviel/ic-light",
-            filename="iclight_sd15_fbc.safetensors",
-            local_dir="./",
-            local_dir_use_symlinks=False
+            filename="iclight_sd15_fbc.safetensors"
         )
     except Exception as e:
         print(f"Failed to download with hf_hub_download: {e}")
@@ -158,16 +151,6 @@ text_encoder = text_encoder.to(device=device, dtype=torch.float16)
 vae = vae.to(device=device, dtype=torch.bfloat16)
 unet = unet.to(device=device, dtype=torch.float16)
 
-# Set attention processors
-if AttnProcessor2_0 is not None:
-    try:
-        unet.set_attn_processor(AttnProcessor2_0())
-        vae.set_attn_processor(AttnProcessor2_0())
-    except Exception as e:
-        print(f"Failed to set attention processors: {e}")
-else:
-    print("AttnProcessor2_0 not available, using default processors")
-
 # Scheduler
 scheduler = DPMSolverMultistepScheduler(
     num_train_timesteps=1000,
@@ -572,10 +555,9 @@ def create_demo():
             fn=process_relight, 
             inputs=inputs, 
             outputs=[result_gallery],
-            show_progress=True,
-            queue=True
+            show_progress=True
         )
-        example_prompts.click(lambda x: x[0], inputs=example_prompts, outputs=prompt, show_progress=False, queue=False)
+        example_prompts.click(lambda x: x[0], inputs=example_prompts, outputs=prompt, show_progress=False)
         
         # Examples - temporarily disabled due to missing image files
         # gr.Examples(