Function open_diffusion_head_source

src/models/vibevoice/lora.cpp:436–450

Opens the diffusion-head fine-tune weights, preferring safetensors over the pickled variants.

// Opens the diffusion-head fine-tune weights, preferring safetensors over the pickled variants.
std::shared_ptr<const assets::TensorSource> open_diffusion_head_source(const std::filesystem::path & adapter_dir) {
    const auto safetensors = adapter_dir / "diffusion_head" / "model.safetensors";
    if (engine::io::is_existing_file(safetensors)) {
        return assets::open_tensor_source(safetensors);
    }
    const auto bin = adapter_dir / "diffusion_head_full.bin";
    if (engine::io::is_existing_file(bin)) {
        return assets::open_torch_bin_tensor_source(bin);
    }
    const auto nested_bin = adapter_dir / "diffusion_head" / "diffusion_head_full.bin";
    if (engine::io::is_existing_file(nested_bin)) {
        return assets::open_torch_bin_tensor_source(nested_bin);
    }
    return nullptr;
}

// Registers full-weight overrides for every adapter tensor that has a matching base tensor. A shape
// mismatch is a hard error (wrong model size); a missing base tensor is skipped like strict=False.