Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 0 additions & 3 deletions neo/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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 \
Expand Down
Loading