Sebastian Krüger
ffbf02b12c
Complete web interface for Meta's AudioCraft AI audio generation: - Gradio UI with tabs for all 5 model families (MusicGen, AudioGen, MAGNeT, MusicGen Style, JASCO) - REST API with FastAPI, OpenAPI docs, and API key auth - VRAM management with ComfyUI coexistence support - SQLite database for project/generation history - Batch processing queue for async generation - Docker deployment optimized for RunPod with RTX 4090 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
78 lines
1.6 KiB
YAML
78 lines
1.6 KiB
YAML
# RunPod Template Configuration
|
|
# Use this as reference when creating a RunPod template
|
|
|
|
name: AudioCraft Studio
|
|
description: AI-powered music and sound generation using Meta's AudioCraft
|
|
|
|
# Container settings
|
|
container:
|
|
image: your-dockerhub-username/audiocraft-studio:latest
|
|
|
|
# Or build from GitHub
|
|
# dockerfile: Dockerfile
|
|
# context: https://github.com/your-username/audiocraft-ui.git
|
|
|
|
# GPU requirements
|
|
gpu:
|
|
type: RTX 4090 # Recommended: RTX 4090, RTX 3090, A100
|
|
count: 1
|
|
minVram: 24 # GB
|
|
|
|
# Resource limits
|
|
resources:
|
|
cpu: 8
|
|
memory: 32 # GB
|
|
disk: 100 # GB (for model cache and outputs)
|
|
|
|
# Port mappings
|
|
ports:
|
|
- name: Gradio UI
|
|
internal: 7860
|
|
external: 7860
|
|
protocol: http
|
|
- name: REST API
|
|
internal: 8000
|
|
external: 8000
|
|
protocol: http
|
|
|
|
# Volume mounts
|
|
volumes:
|
|
- name: outputs
|
|
mountPath: /workspace/outputs
|
|
size: 50 # GB
|
|
- name: models
|
|
mountPath: /workspace/models
|
|
size: 30 # GB (model cache)
|
|
- name: data
|
|
mountPath: /workspace/data
|
|
size: 10 # GB
|
|
|
|
# Environment variables
|
|
env:
|
|
- name: AUDIOCRAFT_HOST
|
|
value: "0.0.0.0"
|
|
- name: AUDIOCRAFT_GRADIO_PORT
|
|
value: "7860"
|
|
- name: AUDIOCRAFT_API_PORT
|
|
value: "8000"
|
|
- name: AUDIOCRAFT_COMFYUI_RESERVE_GB
|
|
value: "10" # Reserve VRAM for ComfyUI if running alongside
|
|
- name: AUDIOCRAFT_MAX_LOADED_MODELS
|
|
value: "2"
|
|
- name: AUDIOCRAFT_IDLE_UNLOAD_MINUTES
|
|
value: "15"
|
|
- name: HF_HOME
|
|
value: "/workspace/models/huggingface"
|
|
|
|
# Startup command
|
|
command: ["python", "main.py"]
|
|
|
|
# Health check
|
|
healthCheck:
|
|
path: /
|
|
port: 7860
|
|
initialDelaySeconds: 120
|
|
periodSeconds: 30
|
|
timeoutSeconds: 10
|
|
failureThreshold: 3
|