diff --git a/neo/Dockerfile b/neo/Dockerfile index 3a120bc..3a4ab86 100644 --- a/neo/Dockerfile +++ b/neo/Dockerfile @@ -68,15 +68,12 @@ ENV PATH="$PATH:${REFORGE_PATH}:/home/ubuntu/.local/bin" # Stop sage from automatically updating due to python version being unknown at build time ENV SAGE_VERSION=2.2.0 ENV SAGE_URL=https://github.com/snw35/sageattention-wheel/releases/download/cu12-${SAGE_VERSION}-cu13-${SAGE_VERSION}/ -ENV SAGE_SHA256=599bede74bb0be04157b24939f02ea0e741d2e9a4e20abc6ae7c98a00c31f4ce -ENV SAGE_UPGRADE=false # # Construct full venv RUN export PYTHON_VERSION=$(python3 ${REFORGE_PATH}/get_python_version.py) \ && export SAGE_CP=$(python3 ${REFORGE_PATH}/get_python_version.py -strip) \ && export SAGE_FILENAME=sageattention-${SAGE_VERSION}+cu13-cp${SAGE_CP}-cp${SAGE_CP}-linux_x86_64.whl \ && curl -L -O ${SAGE_URL}/${SAGE_FILENAME} \ - && echo "$SAGE_SHA256 ./${SAGE_FILENAME}" | sha256sum -c - \ && uv python install $PYTHON_VERSION --default \ && uv venv ${REFORGE_PATH}/venv --clear --seed --allow-existing \ && . ${REFORGE_PATH}/venv/bin/activate \