inference.sh

MODEL_NAME="auffusion/auffusion"
test_data_dir="./data/test_audiocaps.raw.json"  
output_dir="./output/auffusion"
audio_column="spec_path"
caption_column="text"
num_inference_steps=100 
guidance_scale=7.5


training_params="--pretrained_model_name_or_path=$MODEL_NAME \
    --test_data_dir=$test_data_dir \
    --output_dir=$output_dir \
    --audio_column=$audio_column \
    --caption_column=$caption_column \
    --sample_rate=16000 \
    --duration=10 \
    --num_inference_steps=$num_inference_steps \
    --guidance_scale=$guidance_scale \
    --mixed_precision="fp16" \
    --enable_xformers_memory_efficient_attention \
    "

python inference.py $training_params