Building xformers from source

app.py

@@ -4,11 +4,6 @@ from subprocess import getoutput
 # Welcome to the world of python madness!
 
 # Debugging with Verbose mode on...
-os.system('echo "Creating virtual environment..."')
-os.system('python -m venv /home/user/app/venv')
-os.system('source /home/user/app/venv/bin/activate')
-
-
 os.system(f'echo "Checking Python Version"')
 os.system(f"python --version")
 
@@ -23,9 +18,9 @@ os.system(f'echo "Setting working directory to stable-diffusion-webui root folde
 os.chdir("/home/user/app/stable-diffusion-webui")
 
 # 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 "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")
 
 # Patch environment
 os.system(f'echo "Downloading environment patch script..."')
@@ -33,17 +28,17 @@ os.system(f"wget https://github.com/camenduru/webui/raw/main/env_patch.py  -O /h
 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")
+#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")
 
-os.system(f'echo "Removing Training tab from UI..."')
-os.system(f"sed -i -e '/(train_interface, \"Train\", \"ti\"),/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
+#os.system(f'echo "Removing Training tab from UI..."')
+#os.system(f"sed -i -e '/(train_interface, \"Train\", \"ti\"),/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
 
-os.system(f'echo "Removing Extensions tab from UI..."')
-os.system(f"sed -i -e '/extensions_interface, \"Extensions\", \"extensions\"/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
+#os.system(f'echo "Removing Extensions tab from UI..."')
+#os.system(f"sed -i -e '/extensions_interface, \"Extensions\", \"extensions\"/d' /home/user/app/stable-diffusion-webui/modules/ui.py")
 
-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")
+#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..."')
@@ -108,26 +103,34 @@ else:
     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")
 
     # Critical dependency installation sequence
-    os.system(f'echo "starting dependency cleanup..."')
-    os.system(f'pip cache purge')
+    #os.system(f'echo "starting dependency cleanup..."')
+    #os.system(f'pip cache purge')
 
     # Running multiple times to remove stubborn copies
-    os.system(f'echo "Removing all existing numpy versions..."')
-    os.system(f'pip uninstall -y numpy numpy numpy')  
+    #os.system(f'echo "Removing all existing numpy versions..."')
+    #os.system(f'pip uninstall -y numpy numpy numpy')  
     
     # 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")
+    #os.system(f'echo "Installing NumPy 1.26.4..."')
+    #os.system(f"pip install --no-cache-dir numpy==1.26.4")
 
     # Verify NumPy Version 1.26.4
-    os.system(f"python -c 'import numpy; print(numpy.__version__)'")
+    #os.system(f"python -c 'import numpy; print(numpy.__version__)'")
 
     # 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 ")
+    #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 ")
+
+    #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")
+
+    # Building xformers
+    os.system(f'echo "Cloning xformers repo..."')
+    os.system(f'git clone https://github.com/facebookresearch/xformers.git  /home/user/app/xformers')
 
-    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")
+    os.system(f'echo "Installing xformers from source..."')
+    os.chdir(f'/home/user/app/xformers')
+    os.system(f'pip install --no-cache-dir -e .')
 
     # Final verification
     os.system(f'echo "Verifying NumPy, PyTorch, and xformers versions..."')