/src/WasmEdge/include/host/mock/wasmedge_stablediffusion_module.h
Line | Count | Source (jump to first uncovered line) |
1 | | #pragma once |
2 | | |
3 | | #include "host/mock/wasmedge_stablediffusion_func.h" |
4 | | #include "runtime/instance/module.h" |
5 | | |
6 | | namespace WasmEdge { |
7 | | namespace Host { |
8 | | class WasmEdgeStableDiffusionModuleMock |
9 | | : public Runtime::Instance::ModuleInstance { |
10 | | public: |
11 | | WasmEdgeStableDiffusionModuleMock() |
12 | 0 | : Runtime::Instance::ModuleInstance("wasmedge_stablediffusion") { |
13 | 0 | addHostFunc("create_context", |
14 | 0 | std::make_unique<WasmEdgeStableDiffusionMock::CreateContext>()); |
15 | 0 | addHostFunc("text_to_image", |
16 | 0 | std::make_unique<WasmEdgeStableDiffusionMock::TextToImage>()); |
17 | 0 | addHostFunc("image_to_image", |
18 | 0 | std::make_unique<WasmEdgeStableDiffusionMock::ImageToImage>()); |
19 | 0 | addHostFunc("convert", |
20 | 0 | std::make_unique<WasmEdgeStableDiffusionMock::Convert>()); |
21 | 0 | } |
22 | | }; |
23 | | } // namespace Host |
24 | | } // namespace WasmEdge |