Fabio
c36510c21e
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
30 lines
1 KiB
C++
30 lines
1 KiB
C++
#pragma once
|
|
|
|
#include "engine/framework/assets/resource_bundle.h"
|
|
#include "engine/framework/decoders/moss_tts_delay/config.h"
|
|
|
|
#include <cstdint>
|
|
#include <filesystem>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace engine::models::moss_voicegen {
|
|
|
|
// The geometry, token ids and runtimes are the moss_tts_delay family's, shared with
|
|
// the other checkpoints that use this architecture; see
|
|
// engine/framework/decoders/moss_tts_delay/config.h. The aliases keep this model's
|
|
// own names readable at its call sites.
|
|
using MossVoiceGenBackboneConfig = decoders::MossTtsDelayBackboneConfig;
|
|
using MossVoiceGenConfig = decoders::MossTtsDelayConfig;
|
|
|
|
struct MossVoiceGenAssets {
|
|
assets::ResourceBundle resources;
|
|
MossVoiceGenConfig config;
|
|
std::shared_ptr<const assets::TensorSource> model_weights;
|
|
std::shared_ptr<const assets::TensorSource> audio_tokenizer_weights;
|
|
};
|
|
|
|
std::shared_ptr<const MossVoiceGenAssets> load_moss_voicegen_assets(const std::filesystem::path & model_path);
|
|
|
|
} // namespace engine::models::moss_voicegen
|