cpureferencemodel.h

Go to the documentation of this file.

#pragma once
 
#include "transcription/modelpackage.h"
#include "transcription/transcriptionengine.h"
#include "transcription/transcriptiontypes.h"
 
#include <array>
#include <cstdint>
#include <memory>
#include <optional>
 
struct CpuReferenceModelHeader {
    std::array<char, 8> magic{};
    std::uint32_t version = 0;
    std::uint32_t transcriptBytes = 0;
};
 
struct CpuReferenceModelData {
    QString transcript;
};
 
class CpuReferenceModelHandle final : public TranscriptionModelHandle
{
public:
    CpuReferenceModelHandle(ValidatedModelPackage package, CpuReferenceModelData model, qint64 loadTimeMs);
 
    [[nodiscard]] QString backendName() const override;
    [[nodiscard]] ModelMetadata metadata() const override;
    [[nodiscard]] QString modelDescription() const override;
 
    [[nodiscard]] const QString &transcript() const;
 
private:
    ValidatedModelPackage m_package;
    CpuReferenceModelData m_model;
    qint64 m_loadTimeMs = 0;
};
 
[[nodiscard]] std::shared_ptr<const CpuReferenceModelHandle>
loadCpuReferenceModelHandle(const ValidatedModelPackage &package, RuntimeError *error = nullptr);
 
[[nodiscard]] std::shared_ptr<const CpuReferenceModelHandle>
resolveCpuReferenceModelHandle(std::shared_ptr<const TranscriptionModelHandle> model);