| 70 | } |
| 71 | |
| 72 | static void print_usage(const char *name) { |
| 73 | fprintf(stderr, |
| 74 | "Usage: %s -m <models_base_path> -p <prompt> -t <num_threads> [-s <seed> -l <audio_len>]\n\n" |
| 75 | "Options:\n" |
| 76 | " -m <models_base_path> Path to model files\n" |
| 77 | " -p <prompt> Input prompt text (e.g., warm arpeggios on house beats 120BPM with drums effect)\n" |
| 78 | " -t <num_threads> Number of CPU threads to use\n" |
| 79 | " -s <seed> (Optional) Random seed for reproducibility. Different seeds generate different audio samples (Default: %zu)\n" |
| 80 | " -l <audio_len_sec> (Optional) Length of generated audio (Default: %zu s)\n" |
| 81 | " -n <num_steps> (Optional) Number of steps (Default: %zu)\n" |
| 82 | " -o <output_file> (Optional) Output audio file name (Default: <prompt>_<seed>.wav)\n" |
| 83 | " -d <dummy_run> (Optional) Run a dummy run to warm up the model (Default: false)\n" |
| 84 | " -h Show this help message\n", |
| 85 | name, |
| 86 | k_seed_default, |
| 87 | k_audio_len_sec_default, |
| 88 | k_num_steps_default); |
| 89 | } |
| 90 | |
| 91 | static std::string get_filename(std::string prompt, size_t seed) { |
| 92 | // Convert spaces to underscores |