| 5 | #include <sstream> |
| 6 | |
| 7 | SamplingStrategies |
| 8 | SamplingStrategies::from_enum(whisper_sampling_strategy *enum_) { |
| 9 | switch (*enum_) { |
| 10 | case WHISPER_SAMPLING_GREEDY: |
| 11 | return SamplingStrategies(std::make_shared<SamplingGreedy>()); |
| 12 | case WHISPER_SAMPLING_BEAM_SEARCH: |
| 13 | return SamplingStrategies(std::make_shared<SamplingBeamSearch>()); |
| 14 | default: |
| 15 | throw std::runtime_error("Unknown sampling strategy"); |
| 16 | } |
| 17 | }; |
| 18 | |
| 19 | SamplingStrategies |
| 20 | SamplingStrategies::from_sampling_strategy(SamplingType *st) { |