Skip to content

Lora unload

Lora unload #7

Workflow file for this run

name: replicate-ci
on:
pull_request:
branches:
- main
jobs:
push:
runs-on: ubuntu-latest
env:
BASE_MODEL: 'stability-ai/sdxl'
STAGING_MODEL: 'replicate-internal/official-sdxl-staging'
steps:
- name: checkout
uses: actions/checkout@v4
with:
fetch-depth: 0
- name: what changed
id: what-changed
run: |
FILES_CHANGED=$(git diff --name-only --diff-filter=AMR ${{ github.event.before }} ${{ github.event.after }} | xargs)
echo "FILES_CHANGED=$FILES_CHANGED" >> $GITHUB_ENV
if echo "$FILES_CHANGED" | grep -q 'cog.yaml'; then
echo "cog-push=true" >> $GITHUB_OUTPUT
else
echo "cog-push=false" >> $GITHUB_OUTPUT
fi
if ${{ contains(github.event.head_commit.message, '[cog build]') }}; then
echo "cog-push=true" >> $GITHUB_OUTPUT
fi
# if cog.yaml changes - cog build and push. else - yolo build and push!
- name: did-it-tho
env:
COG_PUSH: ${{ steps.what-changed.outputs.cog-push }}
run: |
echo "cog push?: $COG_PUSH"
echo "changed files: $FILES_CHANGED"
- name: setup-cog
if: steps.what-changed.outputs.cog-push == 'true'
uses: replicate/[email protected]
with:
token: ${{ secrets.REPLICATE_API_TOKEN }}
install-cuda: false
- name: cog-build
if: steps.what-changed.outputs.cog-push == 'true'
run: |
cog build --use-cuda-base-image=false
- name: cog-push
if: steps.what-changed.outputs.cog-push == 'true'
env:
REPLICATE_API_TOKEN: ${{secrets.REPLICATE_API_TOKEN}}
run: |
cog push --use-cuda-base-image=false r8.im/"$STAGING_MODEL"
- name: install-yolo
if: steps.what-changed.outputs.cog-push == 'false'
run: |
sudo curl -o /usr/local/bin/yolo -L "https://github.com/replicate/yolo/releases/latest/download/yolo_$(uname -s)_$(uname -m)"
sudo chmod +x /usr/local/bin/yolo
- name: yolo-push
if: steps.what-changed.outputs.cog-push == 'false'
env:
REPLICATE_API_TOKEN: ${{secrets.REPLICATE_API_TOKEN}}
SENTRY_DSN: ${{secrets.SENTRY_DSN}}
run: |
echo "pushing changes from $BASE_MODEL to $STAGING_MODEL"
echo "changed files: $FILES_CHANGED"
yolo push --base $BASE_MODEL --dest $STAGING_MODEL $FILES_CHANGED
- uses: actions/setup-python@v5
with:
python-version: '3.10'
- name: install python test deps
run: |
pip install -r requirements_test.txt
- name: test model
env:
REPLICATE_API_TOKEN: ${{secrets.REPLICATE_API_TOKEN}}
TEST_ENV: 'staging'
run: |
pytest tests/test_predict.py