lowering PyTorch Version and adding Diagnostic checks.

app.py

@@ -90,9 +90,18 @@ 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")
 
+    # Install compatible PyTorch version (2.2.1 works with xformers 0.0.25)
+    os.system(f'echo "Installing compatible PyTorch 2.2.1..."')
+    os.system(f"pip install torch==2.2.1+cu121 torchvision==0.16.2+cu121 --extra-index-url https://download.pytorch.org/whl/cu121 ")
+
     # Install xformers for attention optimization
     os.system(f'echo "Installing xformers for attention optimization..."')
     os.system(f"pip install xformers==0.0.25")
+
+    # diagnostic checks to confirm versions
+    os.system(f'echo "Verifying PyTorch and xformers versions..."')
+    os.system(f"python -c 'import torch; print(torch.__version__)'")
+    os.system(f"python -c 'import xformers; print(xformers.__version__)'")
     
     # Launch WebUI with custom settings
     os.system(f'echo "Starting WebUI with custom configurations..."')