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
26 lines
847 B
C++
26 lines
847 B
C++
#pragma once
|
|
|
|
#include "engine/framework/assets/tensor_source.h"
|
|
#include "engine/models/yue2/types.h"
|
|
|
|
#include <filesystem>
|
|
#include <memory>
|
|
|
|
namespace engine::models::yue2 {
|
|
|
|
struct Yue2Assets {
|
|
std::filesystem::path model_root;
|
|
std::filesystem::path tiktoken_path;
|
|
std::shared_ptr<const assets::TensorSource> model_weights;
|
|
std::shared_ptr<const assets::TensorSource> vae_weights;
|
|
Yue2Config config;
|
|
};
|
|
|
|
std::shared_ptr<const Yue2Assets> load_yue2_assets(const std::filesystem::path & model_path);
|
|
|
|
std::shared_ptr<const assets::TensorSource> make_yue2_lora_source(
|
|
std::shared_ptr<const assets::TensorSource> base,
|
|
const std::filesystem::path & adapter_path, float scale, int64_t layer_count,
|
|
const std::filesystem::path & nar_adapter_path = {}, float nar_scale = 1.0F);
|
|
|
|
} // namespace engine::models::yue2
|