buildx failed with: ERROR: failed to solve: process "/bin/bash -c if [ -n \"$CUDA_VERSION\" ] ; then pip${PYTHON_VERSION} install --upgrade --no-cache-dir ninja==1.11.1 && pip${PYTHON_VERSION} install --upgrade --no-cache-dir --force-reinstall packaging==22.0 && git clone [email protected]:mosaicml/llm-foundry-private.git && cd llm-foundry-private && git checkout evan/no-fwd-hook && pip install .[gpu,openai] && cd .. && git clone https://github.com/dakinggg/composer.git && cd composer && git checkout 2-3-patch && pip install -e .[all] && cd .. && pip install --no-dependencies \"git+ssh://[email protected]/mosaicml/megablocks-private.git@main#egg=megablocks[all]\" && pip install --no-dependencies git+https://github.com/tgale96/[email protected] && pip uninstall -y torch && pip uninstall -y pytorch-triton && pip uninstall -y torchvision && pip install https://download.pytorch.org/whl/nightly/pytorch_triton-2.2.0%2Be28a256d71-cp310-cp310-linux_x86_64.whl && pip install https://download.pytorch.org/whl/nightly/cu121/torch-2.3.0.dev20240110%2Bcu121-cp310-cp310-linux_x86_64.whl && pip install https://download.pytorch.org/whl/nightly/cu121/torchvision-0.18.0.dev20240110%2Bcu121-cp310-cp310-linux_x86_64.whl && git clone --branch v2.4.2 https://github.com/Dao-AILab/flash-attention.git && cd flash-attention && MAX_JOBS=1 python${PYTHON_VERSION} setup.py install && cd .. ; fi" did not complete successfully: exit code: 128