diffusers

setup_local_diffusers.py

@@ -1,7 +1,6 @@
 import os
 import sys
 import subprocess
-import pkg_resources
 
 def setup_local_diffusers():
     # 1. Try to uninstall existing diffusers package
@@ -13,27 +12,21 @@ def setup_local_diffusers():
 
     # 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. Install local diffusers in development mode
-    diffusers_path = os.path.join(current_dir, 'diffusers')
-    if os.path.exists(diffusers_path):
-        try:
-            subprocess.check_call([sys.executable, "-m", "pip", "install", "-e", diffusers_path])
-            print(f"Successfully installed local diffusers from {diffusers_path}")
-        except subprocess.CalledProcessError as e:
-            print(f"Failed to install local diffusers: {e}")
-            return False
-    else:
-        print(f"Error: Could not find diffusers directory at {diffusers_path}")
-        return False
-
-    # 4. Verify the installation
+    # 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}")