testing different pipeline

app.py

@@ -4,32 +4,30 @@ from subprocess import getoutput
 # Welcome to the world of python madness!
 
 # Debugging with Verbose mode on...
-
 os.system(f'echo "Checking Python Version"')
 os.system(f"python --version")
 
-os.system(f'echo "Installing xformers 0.0.15.dev"')
-os.system(f"pip install -q https://huggingface.co/vorstcavry/xformers/resolve/main/xformers-0.0.15.dev0%2B1515f77.d20221130-cp38-cp38-linux_x86_64.whl")
-
 os.system(f'echo "Starting verbose diagnostic setup for stable-diffusion-webui..."')
 
+# Clone repository
 os.system(f'echo "Cloning Stable Diffusion WebUI repository from darkstorm2150..."')
 os.system(f"git clone https://github.com/darkstorm2150/stable-diffusion-webui  /home/user/app/stable-diffusion-webui")
 
+# Set working directory
 os.system(f'echo "Setting working directory to stable-diffusion-webui root folder..."')
 os.chdir("/home/user/app/stable-diffusion-webui")
 
-# Remove numpy/torch from requirements.txt to avoid version conflicts
+# clean requirements.txt
 os.system(f'echo "Removing numpy and torch from requirements.txt to prevent version conflicts..."')
 os.system(f"sed -i '/numpy/d' /home/user/app/stable-diffusion-webui/requirements.txt")
 os.system(f"sed -i '/torch/d' /home/user/app/stable-diffusion-webui/requirements.txt")
 
-os.system(f'echo "Downloading environment patch script from camenduru/webui..."')
+# Patch environment
+os.system(f'echo "Downloading environment patch script..."')
 os.system(f"wget https://github.com/camenduru/webui/raw/main/env_patch.py  -O /home/user/app/env_patch.py")
-
-os.system(f'echo "Patching modules/ui.py to import image_from_url_text from env_patch.py..."')
 os.system(f"sed -i -e '/import image_from_url_text/r /home/user/app/env_patch.py' /home/user/app/stable-diffusion-webui/modules/ui.py")
 
+# UI customization
 os.system(f'echo "Removing Checkpoint Merger tab from UI..."')
 os.system(f"sed -i -e '/(modelmerger_interface, \"Checkpoint Merger\", \"modelmerger\"),/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
 
@@ -42,12 +40,15 @@ os.system(f"sed -i -e '/extensions_interface, \"Extensions\", \"extensions\"/d'
 os.system(f'echo "Removing Settings tab from UI..."')
 os.system(f"sed -i -e '/settings_interface, \"Settings\", \"settings\"/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
 
+# Gradio shadowRoot fix
 os.system(f'echo "Fixing Gradio shadowRoot access in script.js for Hugging Face compatibility..."')
 os.system(f'''sed -i -e "s/document.getElementsByTagName('gradio-app')\\[0\\].shadowRoot/!!document.getElementsByTagName('gradio-app')[0].shadowRoot ? document.getElementsByTagName('gradio-app')[0].shadowRoot : document/g" /home/user/app/stable-diffusion-webui/script.js''')
 
+# Progress bar settings
 os.system(f'echo "Enabling progress bars in UI for better user feedback..."')
 os.system(f"sed -i -e 's/                show_progress=False,/                show_progress=True,/g' /home/user/app/stable-diffusion-webui/modules/ui.py")
 
+# Queue settings
 os.system(f'echo "Modifying webui.py to enable queuing for async processing..."')
 os.system(f"sed -i -e 's/shared.demo.launch/shared.demo.queue().launch/g' /home/user/app/stable-diffusion-webui/webui.py")
 
@@ -101,27 +102,22 @@ else:
     os.system(f'echo "Downloading Protogen X3.4 Stable Diffusion model..."')
     os.system(f"wget -q https://huggingface.co/darkstorm2150/Protogen_x3.4_Official_Release/resolve/main/ProtoGen_X3.4.safetensors  -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/ProtoGen_X3.4.safetensors")
 
-    # Downgrade NumPy
-    os.system(f'echo "Downgrading NumPy to 1.26.4 for compatibility..."')
+    # Critical dependency installation sequence
+    os.system(f'echo "starting dependency cleanup..."')
+    os.system(f'pip cache purge')
+    
+    # Force NumPy 1.26.4 first
+    os.system(f'echo "Installing NumPy 1.26.4..."')
     os.system(f"pip install --no-cache-dir numpy==1.26.4")
 
-    # Install compatible PyTorch version (2.1.2 works with NumPy 1.26.4)
-    os.system(f'echo "Installing PyTorch 2.1.2..."')
+    # Install PyTorch + xformers with CUDA compatibility
+    os.system(f'echo "Installing PyTorch..."')
     os.system(f"pip install torch==2.1.2+cu121 torchvision==0.16.2+cu121 --extra-index-url https://download.pytorch.org/whl/cu121 ")
 
-    # Install xformers
-    os.system(f'echo "Installing xformers for attention optimization..."')
-    os.system(f"pip install xformers==0.0.25")
-
-    # Remove NumPy 2.x (if pulled in by other packages)
-    os.system(f'echo "Removing NumPy 2.x (if present)..."')
-    os.system(f"pip uninstall -y numpy && pip install --no-cache-dir numpy==1.26.4")
-
-    # Reinstall xformers to ensure compatibility
-    os.system(f'echo "Reinstalling xformers to ensure NumPy 1.26.4 compatibility..."')
-    os.system(f"pip uninstall -y xformers && pip install --no-cache-dir xformers==0.0.25")
+    os.system(f'echo "Installing xformers..."')
+    os.system(f"pip install --no-cache-dir xformers==0.0.25 --extra-index-url https://download.pytorch.org/whl/cu118")
 
-    # Diagnostic checks
+    # Final verification
     os.system(f'echo "Verifying NumPy, PyTorch, and xformers versions..."')
     os.system(f"python -c 'import numpy; print(numpy.__version__)'")
     os.system(f"python -c 'import torch; print(torch.__version__)'")