Skip to content

Commit

Permalink
--wip-- [skipci]
Browse files Browse the repository at this point in the history
  • Loading branch information
Liana64 committed Dec 3, 2024
1 parent 5055342 commit 3f822ba
Show file tree
Hide file tree
Showing 8 changed files with 105 additions and 5 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/build-images.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -74,12 +74,12 @@ jobs:
shell: bash
run: |
if [[ -z "${{ inputs.appsToBuild }}" ]]; then
matrices=$(python ./scripts/actions/prepare-matrices.py "all" "${{ inputs.pushImages }}" "${{ inputs.force }}")
matrices=$(python ./scripts/prepare-matrices.py "all" "${{ inputs.pushImages }}" "${{ inputs.force }}")
else
if [[ -z "${{ inputs.channelsToBuild }}" ]]; then
matrices=$(python ./scripts/actions/prepare-matrices.py "${{ inputs.appsToBuild }}" "${{ inputs.pushImages }}" "${{ inputs.force }}")
matrices=$(python ./scripts/prepare-matrices.py "${{ inputs.appsToBuild }}" "${{ inputs.pushImages }}" "${{ inputs.force }}")
else
matrices=$(python ./scripts/actions/prepare-matrices.py "${{ inputs.appsToBuild }}" "${{ inputs.pushImages }}" "${{ inputs.force }}" "${{ inputs.channelsToBuild }}")
matrices=$(python ./scripts/prepare-matrices.py "${{ inputs.appsToBuild }}" "${{ inputs.pushImages }}" "${{ inputs.force }}" "${{ inputs.channelsToBuild }}")
fi
fi
echo "matrices=${matrices}" >> $GITHUB_OUTPUT
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/render-readme.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ jobs:
env:
GITHUB_TOKEN: "${{ steps.app-token.outputs.token }}"
shell: bash
run: python ./scripts/actions/render-readme.py
run: python ./scripts/render-readme.py

- name: Commit Changes
shell: bash
Expand Down
1 change: 0 additions & 1 deletion apps/boltz/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,6 @@ ENV \
USER root
WORKDIR /app

# TODO: Add uv
RUN \
groupadd --gid ${GID} ${USERNAME} \
&& useradd --uid ${UID} --gid ${GID} --create-home --shell /bin/bash ${USERNAME} \
Expand Down
11 changes: 11 additions & 0 deletions apps/qlora/metadata.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
---
#yamllint disable
app: qlora
semver: true
channels:
- name: stable
platforms: ["linux/amd64", "linux/arm64"]
stable: true
tests:
enabled: false
type: cli
79 changes: 79 additions & 0 deletions apps/rfdiffusion/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,79 @@
FROM docker.io/library/python:3.12-slim-bookworm

LABEL \
maintainer="Liana64" \
org.opencontainers.image.source="https://github.com/amelie-iska/RFdiffusion"

ARG TARGETPLATFORM
ARG VERSION
ARG CHANNEL
ARG DEBIAN_FRONTEND=noninteractive

ENV \
NVIDIA_DRIVER_CAPABILITIES="compute,video,utility,graphics" \
PATH="/opt/venv/bin:$PATH" \
UMASK="0002" \
LANG=C.UTF-8 \
TZ="Etc/UTC" \
USERNAME=rare \
UID=900 \
GID=900 \
PYTHONDONTWRITEBYTECODE=1 \
PYTHONUNBUFFERED=1 \
PIP_ROOT_USER_ACTION=ignore \
PIP_NO_CACHE_DIR=1 \
PIP_DISABLE_PIP_VERSION_CHECK=1 \
PIP_BREAK_SYSTEM_PACKAGES=1

USER root
WORKDIR /app

RUN \
groupadd --gid ${GID} ${USERNAME} \
&& useradd --uid ${UID} --gid ${GID} --create-home --shell /bin/bash ${USERNAME} \
&& apt-get update && apt-get install -y --no-install-recommends \
curl \
jq \
unzip \
build-essential \
catatonit \
python3-venv \
&& python3 -m venv /opt/venv \
&& . /opt/venv/bin/activate \
&& curl -o /tmp/Miniconda3-latest-Linux-x86_64.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh \
&& curl -o /tmp/Miniconda3-latest-Linux-aarch64.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-aarch64.sh \
&& curl -fsSL -o /tmp/app.zip "https://github.com/amelie-iska/RFdiffusion/releases/tag/v${VERSION}.zip" \
&& unzip -q /tmp/app.zip -d /app \
&& mv /app/boltz-${VERSION}/LICENSE /app \
&& mv /app/boltz-${VERSION}/README.md /app \
&& mv /app/boltz-${VERSION}/pyproject.toml /app \
&& mv /app/boltz-${VERSION}/docs /app \
&& mv /app/boltz-${VERSION}/examples /app \
&& mv /app/boltz-${VERSION}/scripts /app \
&& mv /app/boltz-${VERSION}/src /app \
&& printf "UpdateMethod=docker\nBranch=master\nPackageVersion=%s\nPackageAuthor=[RareCompute](https://github.com/RareCompute)\n" "${VERSION}" > /app/package_info \
&& pip install --no-cache-dir \
torch \
torchvision \
torchaudio \
&& pip install --no-cache-dir --upgrade . \
&& chown -R ${UID}:${GID} /app && chmod -R 755 /app \
&& chown -R ${UID}:${GID} /opt/venv && chmod -R 755 /opt/venv \
&& apt-get purge -y build-essential \
&& apt-get autoremove -y \
&& apt-get clean \
&& rm -rf /root/.cache /var/lib/apt/lists/* /tmp/* /var/tmp/*

COPY --chown=${UID}:${GID} ./apps/rfdiffusion/entrypoint.sh /entrypoint.sh
RUN chmod -R 755 /entrypoint.sh

USER ${USERNAME}
WORKDIR /cache
VOLUME ["/cache"]

WORKDIR /data
VOLUME ["/data"]

WORKDIR /app

ENTRYPOINT ["/usr/bin/catatonit", "--", "/entrypoint.sh"]
11 changes: 11 additions & 0 deletions apps/rfdiffusion/metadata.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
---
#yamllint disable
app: rfdiffusion
semver: true
channels:
- name: stable
platforms: ["linux/amd64", "linux/arm64"]
stable: true
tests:
enabled: false
type: cli
File renamed without changes.
File renamed without changes.

0 comments on commit 3f822ba

Please sign in to comment.