Skip to content

Commit

Permalink
Building and caching deepspeed
Browse files Browse the repository at this point in the history
  • Loading branch information
Paladinium committed Jan 18, 2025
1 parent f4bc008 commit 3cbbd56
Showing 1 changed file with 30 additions and 0 deletions.
30 changes: 30 additions & 0 deletions .github/workflows/deepspeed.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
name: DeepSpeed Build

on:
workflow_dispatch:
inputs:
tag:
description: 'The tag of the docker image to use for build'
required: true
default: 'paladinium/alltalk_deepspeed:latest'

permissions:
contents: read

jobs:
build:
runs-on: ubuntu-latest

steps:
- name: Run the build process with Docker
uses: addnab/docker-run-action@v3
with:
image: ${{ github.event.inputs.tag }}
options: -v ${{ github.workspace }}/build:/deepspeed
run: |
/build_deepspeed.sh
- name: Upload whl as artifact
uses: actions/upload-artifact@v4
with:
name: 'deepspeed-whl'
path: ${{ github.workspace }}/build/*.whl

0 comments on commit 3cbbd56

Please sign in to comment.