Skip to content

Commit

Permalink
chore: update alpha to beta phase (#203)
Browse files Browse the repository at this point in the history
This commit updates the code and documentation to signal we are entering
the Beta phase of the AI network journey.
  • Loading branch information
rickstaa authored Sep 17, 2024
1 parent 99a64c9 commit af9bba3
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 13 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# ai-worker

> [!WARNING]
> This is a prototype repository undergoing rapid changes. It's not intended for production use yet.
> The AI network is in it's **Beta** phase and although it is ready for production it is still under development. Please report any issues you encounter to the [Livepeer Discord](https://discord.gg/7nbPbTK).
This repository hosts the AI worker and runner for processing inference requests on the Livepeer AI subnet.

Expand Down
4 changes: 2 additions & 2 deletions runner/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -56,11 +56,11 @@ The AI Runner container's runner app uses [HuggingFace](https://huggingface.co/)
3. **Download AI Models**: Use the [ld_checkpoints.sh](https://github.com/livepeer/ai-worker/blob/main/runner/dl_checkpoints.sh) script to download models from `aiModels.json` to `~/.lpData/models`. Run the following command in the `.lpData` directory:

```bash
curl -s https://raw.githubusercontent.com/livepeer/ai-worker/main/runner/dl_checkpoints.sh | bash -s -- --alpha
curl -s https://raw.githubusercontent.com/livepeer/ai-worker/main/runner/dl_checkpoints.sh | bash -s -- --beta
```

> [!NOTE]
> The `--alpha` flag downloads only models supported by the Livepeer.inc Gateway node on the _AI Subnet_. Remove this flag to download all models.
> The `--beta` flag downloads only models supported by the Livepeer.inc Gateway node on the _AI Subnet_. Remove this flag to download all models.
> [!TIP]
> If you want to download individual models you can checkout the `dl-checkpoints.sh` script to see how to download model checkpoints to the local `models` directory.
Expand Down
20 changes: 10 additions & 10 deletions runner/dl_checkpoints.sh
Original file line number Diff line number Diff line change
Expand Up @@ -11,16 +11,16 @@ check_hf_auth() {
# Displays help message.
function display_help() {
echo "Description: This script is used to download models available on the Livepeer AI Subnet."
echo "Usage: $0 [--alpha]"
echo "Usage: $0 [--beta]"
echo "Options:"
echo " --alpha Download alpha models."
echo " --beta Download beta models."
echo " --restricted Download models with a restrictive license."
echo " --help Display this help message."
}

# Download recommended models during alpha phase.
function download_alpha_models() {
printf "\nDownloading recommended alpha phase models...\n"
# Download recommended models during beta phase.
function download_beta_models() {
printf "\nDownloading recommended beta phase models...\n"

printf "\nDownloading unrestricted models...\n"

Expand All @@ -44,7 +44,7 @@ function download_alpha_models() {

# Download all models.
function download_all_models() {
download_alpha_models
download_beta_models

printf "\nDownloading other available models...\n"

Expand Down Expand Up @@ -87,8 +87,8 @@ MODE="all"
for arg in "$@"
do
case $arg in
--alpha)
MODE="alpha"
--beta)
MODE="beta"
shift
;;
--restricted)
Expand Down Expand Up @@ -116,8 +116,8 @@ if ! command -v huggingface-cli > /dev/null 2>&1; then
exit 1
fi

if [ "$MODE" = "alpha" ]; then
download_alpha_models
if [ "$MODE" = "beta" ]; then
download_beta_models
elif [ "$MODE" = "restricted" ]; then
download_restricted_models
else
Expand Down

0 comments on commit af9bba3

Please sign in to comment.