| 21 | std::string model_name = request.model_path.filename().generic_string(); |
| 22 | std::transform(model_name.begin(), model_name.end(), model_name.begin(), [](unsigned char ch) { |
| 23 | return static_cast<char>(std::tolower(ch)); |
| 24 | }); |
| 25 | if (model_name.find("base") != std::string::npos) { |
| 26 | selection.dit_model_path = "acestep-v15-base"; |
| 27 | } else if (model_name.find("turbo") != std::string::npos) { |
| 28 | selection.dit_model_path = "acestep-v15-turbo"; |
| 29 | } |
| 30 | return selection; |
| 31 | } |
| 32 | |
| 33 | runtime::ModelMetadata metadata(const AceStepAssets & assets) { |
| 34 | runtime::ModelMetadata out; |
| 35 | out.family = "ace_step"; |
| 36 | out.variant = assets.selection.dit_model_path; |
| 37 | out.description = "ACE-Step loaded from local extracted assets."; |
| 38 | return out; |
| 39 | } |
| 40 | |
| 41 | runtime::CapabilitySet capabilities(const AceStepAssets &) { |
no test coverage detected