Some checks failed
Linux Build / Linux cpu (push) Has been cancelled
Linux Build / Linux cpu-clang (push) Has been cancelled
Linux Build / Linux vulkan (push) Has been cancelled
mac-build / macOS CPU (push) Has been cancelled
Nix Build / Nix (cpu) (push) Has been cancelled
Nix Build / Nix (vulkan) (push) Has been cancelled
windows-build / Windows CPU (push) Has been cancelled
33 lines
866 B
C++
33 lines
866 B
C++
#pragma once
|
|
|
|
#include "engine/models/magpie_tts/assets.h"
|
|
#include "engine/models/magpie_tts/types.h"
|
|
|
|
#include <filesystem>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace engine::models::magpie_tts {
|
|
|
|
class MagpieTextTokenizer {
|
|
public:
|
|
explicit MagpieTextTokenizer(std::filesystem::path resource_root);
|
|
~MagpieTextTokenizer();
|
|
|
|
MagpieTextTokenizer(const MagpieTextTokenizer &) = delete;
|
|
MagpieTextTokenizer & operator=(const MagpieTextTokenizer &) = delete;
|
|
|
|
MagpieTokenizationResult tokenize(
|
|
const std::string & text,
|
|
const MagpieTTSGenerationOptions & options) const;
|
|
|
|
static std::string normalize_language(std::string language);
|
|
static std::vector<std::string> supported_native_languages();
|
|
|
|
private:
|
|
class Impl;
|
|
std::unique_ptr<Impl> impl_;
|
|
};
|
|
|
|
} // namespace engine::models::magpie_tts
|