Sebastian Krüger
9ee626a78e
Major architecture overhaul to address RunPod Docker limitations: Core Infrastructure: - Add base_service.py: Abstract base class for all AI services - Add service_manager.py: Process lifecycle management - Add core/requirements.txt: Core dependencies Model Services (Standalone Python): - Add models/vllm/server.py: Qwen 2.5 7B text generation - Add models/flux/server.py: Flux.1 Schnell image generation - Add models/musicgen/server.py: MusicGen Medium music generation - Each service inherits from GPUService base class - OpenAI-compatible APIs - Standalone execution support Ansible Deployment: - Add playbook.yml: Comprehensive deployment automation - Add ansible.cfg: Ansible configuration - Add inventory.yml: Localhost inventory - Tags: base, python, dependencies, models, tailscale, validate, cleanup Scripts: - Add scripts/install.sh: Full installation wrapper - Add scripts/download-models.sh: Model download wrapper - Add scripts/start-all.sh: Start orchestrator - Add scripts/stop-all.sh: Stop all services Documentation: - Update ARCHITECTURE.md: Document distributed VPS+GPU architecture Benefits: - No Docker: Avoids RunPod CAP_SYS_ADMIN limitations - Fully reproducible via Ansible - Extensible: Add models in 3 steps - Direct Python execution (no container overhead) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
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 "========================================="
|