37 lines
687 B
Bash
37 lines
687 B
Bash
|
#!/bin/bash
|
||
|
|
#
|
||
|
|
# Download AI Models
|
||
|
|
# Wrapper for Ansible models tag
|
||
|
|
#
|
||
|
|
|
||
|
|
set -e
|
||
|
|
|
||
|
|
cd "$(dirname "$0")/.."
|
||
|
|
|
||
|
|
echo "========================================="
|
||
|
|
echo " Downloading AI Models (~37GB)"
|
||
|
|
echo "========================================="
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
# Source .env if it exists
|
||
|
|
if [ -f .env ]; then
|
||
|
|
set -a
|
||
|
|
source .env
|
||
|
|
set +a
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Check HF_TOKEN
|
||
|
|
if [ -z "$HF_TOKEN" ]; then
|
||
|
|
echo "Error: HF_TOKEN not set"
|
||
|
|
echo "Add HF_TOKEN to .env file"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Run Ansible with models tag
|
||
|
|
ansible-playbook playbook.yml --tags models
|
||
|
|
|
||
|
|
echo ""
|
||
|
|
echo "========================================="
|
||
|
|
echo " Model download complete!"
|
||
|
|
echo "========================================="
|