diffusers

app.py

@@ -1,5 +1,4 @@
-import setup_local_diffusers
-setup_local_diffusers.setup_local_diffusers()
+
 import spaces
 import gradio as gr
 from gradio_imageslider import ImageSlider
@@ -13,9 +12,6 @@ from module.ip_adapter.utils import load_adapter_to_pipe
 from pipelines.sdxl_instantir import InstantIRPipeline
 import os
 
-import os
-import sys
-
 os.makedirs('./models', exist_ok=True)
 # Download model files if not present
 for filename in ["adapter.pt", "aggregator.pt", "previewer_lora_weights.bin"]:

module/aggregator.py

@@ -6,7 +6,7 @@ from torch import nn
 from torch.nn import functional as F
 
 from diffusers.configuration_utils import ConfigMixin, register_to_config
-from diffusers.loaders.single_file_model import FromOriginalModelMixin
+from diffusers.loaders.single_file import FromOriginalModelMixin
 from diffusers.utils import BaseOutput, logging
 from diffusers.models.attention_processor import (
     ADDED_KV_ATTENTION_PROCESSORS,

setup_local_diffusers.py

@@ -1,40 +0,0 @@
-import os
-import sys
-import subprocess
-
-def setup_local_diffusers():
-    # 1. Try to uninstall existing diffusers package
-    try:
-        subprocess.check_call([sys.executable, "-m", "pip", "uninstall", "diffusers", "-y"])
-        print("Successfully uninstalled existing diffusers package")
-    except subprocess.CalledProcessError:
-        print("No existing diffusers package found to uninstall")
-
-    # 2. Get the current directory and add it to PYTHONPATH
-    current_dir = os.path.dirname(os.path.abspath(__file__))
-    
-    # Add the parent directory of diffusers to Python path
-    if current_dir not in sys.path:
-        sys.path.insert(0, current_dir)
-        print(f"Added {current_dir} to PYTHONPATH")
-
-    # 3. Verify the import works
-    try:
-        import diffusers
-        print(f"Successfully imported diffusers from: {diffusers.__file__}")
-        
-        # Verify specific module exists
-        from diffusers.loaders.single_file import FromOriginalModelMixin
-        print("Successfully imported FromOriginalModelMixin")
-        
-        return True
-    except ImportError as e:
-        print(f"Failed to import diffusers: {e}")
-        return False
-
-if __name__ == "__main__":
-    success = setup_local_diffusers()
-    if success:
-        print("Local diffusers setup completed successfully")
-    else:
-        print("Failed to setup local diffusers")