Update Dockerfile

Dockerfile

@@ -43,5 +43,11 @@ EXPOSE 7860
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 ENV GRADIO_SERVER_PORT="7860"
 
+# --- NEW MEMORY OPTIMIZATIONS ---
+# 1. Tells PyTorch to manage memory more efficiently to prevent fragmentation
+ENV PYTORCH_CUDA_ALLOC_CONF="expandable_segments:True"
+# 2. Tells Pixal3D to use its built-in low VRAM mode (caps resolution at 1024 instead of 1536)
+ENV LOW_VRAM="1"
+
 # Run the application
 CMD ["python", "app.py"]