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
28 lines
1 KiB
C++
28 lines
1 KiB
C++
#include "engine/community_models/glm_tts/assets.h"
|
|
|
|
#include <exception>
|
|
#include <iostream>
|
|
|
|
int main(int argc, char ** argv) try {
|
|
if (argc != 2) {
|
|
std::cerr << "usage: glm_tts_assets_probe <model-path>\n";
|
|
return 2;
|
|
}
|
|
const auto assets =
|
|
engine::models::glm_tts::load_glm_tts_assets(argv[1]);
|
|
const auto & config = assets->config;
|
|
std::cout
|
|
<< "{\"llama_layers\":" << config.llama.num_hidden_layers
|
|
<< ",\"llama_hidden\":" << config.llama.hidden_size
|
|
<< ",\"speech_layers\":" << config.speech_tokenizer.pooling_position
|
|
<< ",\"speech_vocab\":" << config.speech_tokenizer.quantize_vocab_size
|
|
<< ",\"flow_depth\":" << config.flow.depth
|
|
<< ",\"sample_rate\":" << config.sample_rate
|
|
<< ",\"audio_token_start\":" << config.audio_token_start
|
|
<< ",\"audio_token_end\":" << config.audio_token_end
|
|
<< "}\n";
|
|
return 0;
|
|
} catch (const std::exception & error) {
|
|
std::cerr << "glm_tts_assets_probe failed: " << error.what() << '\n';
|
|
return 1;
|
|
}
|