Attempt to fix Exit code 137 (OOM) by using --no-build-isolation for transformer-engine

Dockerfile

@@ -61,7 +61,8 @@ RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
 # Install Transformer Engine separately after PyTorch and cuDNN are in place and headers are linked.
 RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
     conda activate cosmos-predict1 && \
-    pip install --no-cache-dir transformer-engine[pytorch]==1.12.0
+    #pip install transformer-engine[pytorch]==1.12.0
+    pip install --no-cache-dir --no-build-isolation transformer-engine[pytorch]==1.12.0
 
 # --- Verification Steps ---
 RUN echo "Verifying Python and Conda installations..."