| 41 | } |
| 42 | |
| 43 | int bark_params_parse(int argc, char** argv, bark_params& params) { |
| 44 | for (int i = 1; i < argc; i++) { |
| 45 | std::string arg = argv[i]; |
| 46 | |
| 47 | if (arg == "-t" || arg == "--threads") { |
| 48 | params.n_threads = std::stoi(argv[++i]); |
| 49 | } else if (arg == "-p" || arg == "--prompt") { |
| 50 | params.prompt = argv[++i]; |
| 51 | } else if (arg == "-m" || arg == "--model_path") { |
| 52 | params.model_path = argv[++i]; |
| 53 | } else if (arg == "-s" || arg == "--seed") { |
| 54 | params.seed = std::stoi(argv[++i]); |
| 55 | } else if (arg == "-o" || arg == "--outwav") { |
| 56 | params.dest_wav_path = argv[++i]; |
| 57 | } else if (arg == "-h" || arg == "--help") { |
| 58 | bark_print_usage(argv, params); |
| 59 | exit(0); |
| 60 | } else { |
| 61 | fprintf(stderr, "error: unknown argument: %s\n", arg.c_str()); |
| 62 | bark_print_usage(argv, params); |
| 63 | exit(0); |
| 64 | } |
| 65 | } |
| 66 | |
| 67 | return 0; |
| 68 | } |
nothing calls this directly
no test coverage detected