branch-finetune-test.yaml

name: Branch fine-tuning test

on:
  workflow_dispatch:
    inputs:
      model:
        description: "Model"
        required: true
        type: string
      branch:
        description: "Branch"
        required: true
        type: string

  workflow_call:
    inputs:
      model:
        required: true
        type: string
      branch:
        required: true
        type: string

permissions:
  contents: read

jobs:
  finetune_inference:
    runs-on: self-hosted

    steps:
      - name: get repo
        run: |
          rm -rf $HOME/mLoRA
          git clone https://github.com/TUDB-Labs/mLoRA $HOME/mLoRA
      - name: checkout branch
        run: |
          cd $HOME/mLoRA
          git fetch origin ${{ inputs.branch }}:local_tmp
          git switch local_tmp
      - name: install dependencies
        run: |
          cd $HOME/mLoRA
          python -m pip install --upgrade pip
          pip install .[test]
      - name: finetune all test case
        run: |
          cd $HOME/mLoRA
          tests/finetune_all_case.sh ${{ inputs.model }}