weightsfile.hpp

#pragma once
 
#include "../model/model.hpp"
 
#include <filesystem>
#include <istream>
#include <string>
 
namespace dl::io {
 
    class WeightsFileFormat {
    public:
        virtual ~WeightsFileFormat() = default;
 
        virtual bool loadModelFromStream(dl::ModelBase& model, std::istream& stream) = 0;
 
        virtual bool loadModelFromFile(dl::ModelBase& model, std::filesystem::path& path, bool mmap = true) = 0;
 
        virtual void writeModelToStream(dl::ModelBase& model, std::ostream& stream) = 0;
    };
 
    extern dl::io::WeightsFileFormat& safetensorsFormat;
    extern dl::io::WeightsFileFormat& ggufFormat;
 
} // namespace dl::io