stable_diffusion.hpp

Go to the documentation of this file.

#ifndef ACMXVK_STABLE_DIFFUSION_HPP
#define ACMXVK_STABLE_DIFFUSION_HPP
 
#include <opencv2/core.hpp>
 
#include <cstdint>
#include <filesystem>
#include <functional>
#include <string>
#include <vector>
 
namespace acmxvk::stable_diffusion {
    struct Lora {
        std::filesystem::path path;
        double multiplier = 1.0;
    };
 
    struct Settings {
        std::filesystem::path server_executable;
        std::filesystem::path model;
        std::filesystem::path upscale_model;
        std::filesystem::path lora_directory;
        std::vector<Lora> loras;
        std::vector<std::string> server_arguments;
        std::string prompt;
        std::string negative_prompt;
        std::string sampler;
        std::string scheduler;
        int width = 576;
        int height = 320;
        int upscale_width = 0;
        int upscale_height = 0;
        int upscale_working_width = 0;
        int upscale_working_height = 0;
        int steps = 12;
        int seed = 1234;
        int port = 1234;
        double strength = 0.35;
        double cfg_scale = 5.0;
        bool resize_to_input = true;
        bool upscale_only = false;
        bool quiet = false;
        std::function<bool()> cancelled;
    };
 
    class Server final {
      public:
        explicit Server(Settings settings);
        ~Server();
 
        Server(const Server &) = delete;
        Server &operator=(const Server &) = delete;
        Server(Server &&) = delete;
        Server &operator=(Server &&) = delete;
 
        [[nodiscard]] cv::Mat process(const cv::Mat &rgba) const;
 
      private:
        Settings settings;
        std::int64_t process_id = -1;
#ifdef _WIN32
        std::intptr_t process_handle = 0;
#endif
        std::string endpoint;
        std::filesystem::path diagnostic_log_path;
 
        void start();
        void stop() noexcept;
        void waitUntilReady();
        void resetDiagnosticLog() const noexcept;
        [[nodiscard]] std::string diagnosticLogDetails() const;
    };
} // namespace acmxvk::stable_diffusion
 
#endif