# Vocence chute config for Higgs Audio v2 (example_repo/miner.py) — build-time only.
# Generation + tokenizer weights: HF snapshot (repo) at runtime; image installs PyTorch + Transformers.

Image:
  from_base: parachutes/base-python:3.12.9
  set_user: root
  run_command:
    # Optional if you use processor.save_audio / soundfile-backed I/O elsewhere.
    - apt-get update && apt-get install -y --no-install-recommends libsndfile1 && rm -rf /var/lib/apt/lists/*
    # Match your Chutes CUDA major: cu124 → cu126 / cu128 on pytorch.org if needed.
    - pip install --no-cache-dir  "transformers>=5.3.0"
    - pip install --no-cache-dir "transformers>=4.53.0" accelerate huggingface_hub safetensors pyyaml soundfile
    - pip install --no-cache-dir numpy einops scipy tqdm protobuf sentencepiece filelock packaging regex requests

NodeSelector:
  gpu_count: 1
  min_vram_gb_per_gpu: 24
  include: ['pro_6000']
  
Chute:
  tagline: Vocence PromptTTS — Higgs Audio v2 (miner.py)
  readme: Higgs Audio v2 + Vocence Miner; device_map=auto. Set HIGGS_AUDIO_TOKENIZER_REPO or bundle tokenizer under the HF repo if not using default Hub tokenizer.
  shutdown_after_seconds: 86400
  concurrency: 1
  max_instances: 1
  scaling_threshold: 0.5
  tee: true