Skip to content

Commit

Permalink
chore: added model conversion for llama2.c
Browse files Browse the repository at this point in the history
  • Loading branch information
limcheekin committed Sep 6, 2023
1 parent 4c09b66 commit 11e82ab
Showing 1 changed file with 28 additions and 0 deletions.
28 changes: 28 additions & 0 deletions .github/workflows/convert.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
# REF: https://docs.github.com/en/actions/using-github-hosted-runners/about-github-hosted-runners
name: Convert models and upload to HF
on:
workflow_dispatch:
inputs:
model_name:
required: true
description: Model name of hugging face such as PY007/TinyLlama-1.1B-step-50K-105b.

jobs:
convert-upload:
runs-on: ubuntu-latest
steps:
- name: Checkout own repo
uses: actions/checkout@v3

- name: Setup Python
uses: actions/setup-python@v4
with:
python-version: "3.10"

- name: Install Python packages
run: |
cd ${{ github.ref_name }}
pip install --no-cache-dir -r requirements.txt
- name: Download and convert model
run: cd ${{ github.ref_name }};python export.py tinyllama.bin --hf ${{ github.event.inputs.model_name }}

0 comments on commit 11e82ab

Please sign in to comment.