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
|