@nodetool-ai/runtime 0.8.4-a24.2 → 0.8.4

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (101) hide show
  1. package/dist/.nodetool-build-stamp +1 -1
  2. package/dist/index.d.ts +1 -0
  3. package/dist/index.d.ts.map +1 -1
  4. package/dist/index.js +1 -0
  5. package/dist/index.js.map +1 -1
  6. package/dist/model-load-events.d.ts +38 -0
  7. package/dist/model-load-events.d.ts.map +1 -0
  8. package/dist/model-load-events.js +47 -0
  9. package/dist/model-load-events.js.map +1 -0
  10. package/dist/providers/anthropic-provider.d.ts.map +1 -1
  11. package/dist/providers/anthropic-provider.js +9 -2
  12. package/dist/providers/anthropic-provider.js.map +1 -1
  13. package/dist/providers/chatgpt-provider.d.ts +16 -4
  14. package/dist/providers/chatgpt-provider.d.ts.map +1 -1
  15. package/dist/providers/chatgpt-provider.js +69 -18
  16. package/dist/providers/chatgpt-provider.js.map +1 -1
  17. package/dist/providers/cohere-provider.d.ts.map +1 -1
  18. package/dist/providers/cohere-provider.js +9 -21
  19. package/dist/providers/cohere-provider.js.map +1 -1
  20. package/dist/providers/cost-calculator.d.ts.map +1 -1
  21. package/dist/providers/cost-calculator.js +5 -0
  22. package/dist/providers/cost-calculator.js.map +1 -1
  23. package/dist/providers/embedding-http.d.ts +21 -0
  24. package/dist/providers/embedding-http.d.ts.map +1 -0
  25. package/dist/providers/embedding-http.js +38 -0
  26. package/dist/providers/embedding-http.js.map +1 -0
  27. package/dist/providers/fake-app-greeting.d.ts +6 -0
  28. package/dist/providers/fake-app-greeting.d.ts.map +1 -0
  29. package/dist/providers/fake-app-greeting.js +139 -0
  30. package/dist/providers/fake-app-greeting.js.map +1 -0
  31. package/dist/providers/fake-provider.d.ts +6 -1
  32. package/dist/providers/fake-provider.d.ts.map +1 -1
  33. package/dist/providers/fake-provider.js +43 -14
  34. package/dist/providers/fake-provider.js.map +1 -1
  35. package/dist/providers/gemini-web-provider.d.ts +2 -0
  36. package/dist/providers/gemini-web-provider.d.ts.map +1 -1
  37. package/dist/providers/gemini-web-provider.js +10 -16
  38. package/dist/providers/gemini-web-provider.js.map +1 -1
  39. package/dist/providers/gguf-header.d.ts +10 -0
  40. package/dist/providers/gguf-header.d.ts.map +1 -0
  41. package/dist/providers/gguf-header.js +97 -0
  42. package/dist/providers/gguf-header.js.map +1 -0
  43. package/dist/providers/hf-hub-file.d.ts +3 -0
  44. package/dist/providers/hf-hub-file.d.ts.map +1 -0
  45. package/dist/providers/hf-hub-file.js +42 -0
  46. package/dist/providers/hf-hub-file.js.map +1 -0
  47. package/dist/providers/index.d.ts +4 -0
  48. package/dist/providers/index.d.ts.map +1 -1
  49. package/dist/providers/index.js +13 -6
  50. package/dist/providers/index.js.map +1 -1
  51. package/dist/providers/jina-provider.d.ts +0 -5
  52. package/dist/providers/jina-provider.d.ts.map +1 -1
  53. package/dist/providers/jina-provider.js +9 -29
  54. package/dist/providers/jina-provider.js.map +1 -1
  55. package/dist/providers/mlx-models.d.ts +33 -0
  56. package/dist/providers/mlx-models.d.ts.map +1 -0
  57. package/dist/providers/mlx-models.js +64 -0
  58. package/dist/providers/mlx-models.js.map +1 -0
  59. package/dist/providers/node-llama-cpp-models.d.ts +28 -0
  60. package/dist/providers/node-llama-cpp-models.d.ts.map +1 -0
  61. package/dist/providers/node-llama-cpp-models.js +82 -0
  62. package/dist/providers/node-llama-cpp-models.js.map +1 -0
  63. package/dist/providers/node-llama-cpp-provider.d.ts +9 -2
  64. package/dist/providers/node-llama-cpp-provider.d.ts.map +1 -1
  65. package/dist/providers/node-llama-cpp-provider.js +62 -13
  66. package/dist/providers/node-llama-cpp-provider.js.map +1 -1
  67. package/dist/providers/node-llama-cpp-trending.generated.d.ts +3 -0
  68. package/dist/providers/node-llama-cpp-trending.generated.d.ts.map +1 -0
  69. package/dist/providers/node-llama-cpp-trending.generated.js +253 -0
  70. package/dist/providers/node-llama-cpp-trending.generated.js.map +1 -0
  71. package/dist/providers/openai-provider.d.ts.map +1 -1
  72. package/dist/providers/openai-provider.js +8 -11
  73. package/dist/providers/openai-provider.js.map +1 -1
  74. package/dist/providers/python-provider.d.ts +2 -0
  75. package/dist/providers/python-provider.d.ts.map +1 -1
  76. package/dist/providers/python-provider.js +13 -4
  77. package/dist/providers/python-provider.js.map +1 -1
  78. package/dist/providers/sdcpp-models.d.ts +60 -0
  79. package/dist/providers/sdcpp-models.d.ts.map +1 -0
  80. package/dist/providers/sdcpp-models.js +435 -0
  81. package/dist/providers/sdcpp-models.js.map +1 -0
  82. package/dist/providers/sdcpp-provider.d.ts +40 -0
  83. package/dist/providers/sdcpp-provider.d.ts.map +1 -0
  84. package/dist/providers/sdcpp-provider.js +353 -0
  85. package/dist/providers/sdcpp-provider.js.map +1 -0
  86. package/dist/providers/sdcpp-server.d.ts +13 -0
  87. package/dist/providers/sdcpp-server.d.ts.map +1 -0
  88. package/dist/providers/sdcpp-server.js +150 -0
  89. package/dist/providers/sdcpp-server.js.map +1 -0
  90. package/dist/providers/text-tool-calls.d.ts +52 -0
  91. package/dist/providers/text-tool-calls.d.ts.map +1 -0
  92. package/dist/providers/text-tool-calls.js +141 -0
  93. package/dist/providers/text-tool-calls.js.map +1 -0
  94. package/dist/providers/types.d.ts +11 -0
  95. package/dist/providers/types.d.ts.map +1 -1
  96. package/dist/providers/types.js.map +1 -1
  97. package/dist/providers/voyage-provider.d.ts +0 -5
  98. package/dist/providers/voyage-provider.d.ts.map +1 -1
  99. package/dist/providers/voyage-provider.js +9 -29
  100. package/dist/providers/voyage-provider.js.map +1 -1
  101. package/package.json +5 -5
@@ -0,0 +1,435 @@
1
+ /**
2
+ * The models the stable-diffusion.cpp provider offers. A model loads its
3
+ * files from one or more Hugging Face repositories. Text encoders and VAEs
4
+ * that several models share come from one repository and file, so they
5
+ * download once.
6
+ */
7
+ export const SD_CPP_COMPONENT_FLAGS = {
8
+ model: "--model",
9
+ diffusion_model: "--diffusion-model",
10
+ uncond_diffusion_model: "--uncond-diffusion-model",
11
+ vae: "--vae",
12
+ clip_l: "--clip_l",
13
+ clip_g: "--clip_g",
14
+ t5xxl: "--t5xxl",
15
+ llm: "--llm",
16
+ llm_vision: "--llm_vision",
17
+ tokenizer: "--tokenizer",
18
+ embeddings_connectors: "--embeddings-connectors"
19
+ };
20
+ const hf = (repoId, path) => ({ repoId, path });
21
+ // Shared text encoders and VAEs.
22
+ const FLUX1_VAE = hf("second-state/FLUX.1-schnell-GGUF", "ae.safetensors");
23
+ const FLUX1_CLIP_L = hf("second-state/FLUX.1-schnell-GGUF", "clip_l-Q8_0.gguf");
24
+ const FLUX1_T5XXL = hf("second-state/FLUX.1-schnell-GGUF", "t5xxl-Q4_0.gguf");
25
+ const FLUX2_VAE = hf("Comfy-Org/flux2-dev", "split_files/vae/flux2-vae.safetensors");
26
+ const QWEN_IMAGE_VAE = hf("Comfy-Org/Qwen-Image_ComfyUI", "split_files/vae/qwen_image_vae.safetensors");
27
+ const QWEN3_4B = hf("unsloth/Qwen3-4B-GGUF", "Qwen3-4B-Q4_K_M.gguf");
28
+ const QWEN3_8B = hf("unsloth/Qwen3-8B-GGUF", "Qwen3-8B-Q4_K_M.gguf");
29
+ // Also the llama.cpp provider's recommended chat model.
30
+ const QWEN3_4B_2507 = hf("unsloth/Qwen3-4B-Instruct-2507-GGUF", "Qwen3-4B-Instruct-2507-Q4_K_M.gguf");
31
+ const QWEN3_VL_4B = hf("Qwen/Qwen3-VL-4B-Instruct-GGUF", "Qwen3VL-4B-Instruct-Q4_K_M.gguf");
32
+ const QWEN3_VL_4B_VISION = hf("Qwen/Qwen3-VL-4B-Instruct-GGUF", "mmproj-Qwen3VL-4B-Instruct-F16.gguf");
33
+ const QWEN3_VL_8B = hf("Qwen/Qwen3-VL-8B-Instruct-GGUF", "Qwen3VL-8B-Instruct-Q4_K_M.gguf");
34
+ const QWEN3_VL_8B_VISION = hf("Qwen/Qwen3-VL-8B-Instruct-GGUF", "mmproj-Qwen3VL-8B-Instruct-F16.gguf");
35
+ const QWEN25_VL_7B = hf("mradermacher/Qwen2.5-VL-7B-Instruct-GGUF", "Qwen2.5-VL-7B-Instruct.Q4_K_M.gguf");
36
+ const QWEN25_VL_7B_VISION = hf("mradermacher/Qwen2.5-VL-7B-Instruct-GGUF", "Qwen2.5-VL-7B-Instruct.mmproj-Q8_0.gguf");
37
+ export const SD_CPP_MODELS = [
38
+ {
39
+ id: "sd-turbo",
40
+ name: "SD-Turbo",
41
+ files: { model: hf("Green-Sky/SD-Turbo-GGUF", "sd_turbo-f16-q8_0.gguf") },
42
+ sizeGb: 2,
43
+ imageInput: "init",
44
+ defaults: { steps: 4, cfgScale: 1, size: 512 }
45
+ },
46
+ {
47
+ id: "sd-1.5",
48
+ name: "Stable Diffusion 1.5",
49
+ files: {
50
+ model: hf("second-state/stable-diffusion-v1-5-GGUF", "stable-diffusion-v1-5-pruned-emaonly-Q8_0.gguf")
51
+ },
52
+ sizeGb: 1.8,
53
+ imageInput: "init",
54
+ defaults: { steps: 20, cfgScale: 7, size: 512 }
55
+ },
56
+ {
57
+ id: "sd-3.5-medium",
58
+ name: "Stable Diffusion 3.5 Medium",
59
+ files: {
60
+ model: hf("second-state/stable-diffusion-3.5-medium-GGUF", "sd3.5_medium-Q4_0.gguf"),
61
+ clip_l: hf("second-state/stable-diffusion-3.5-medium-GGUF", "clip_l-Q8_0.gguf"),
62
+ clip_g: hf("second-state/stable-diffusion-3.5-medium-GGUF", "clip_g-Q8_0.gguf"),
63
+ t5xxl: hf("second-state/stable-diffusion-3.5-medium-GGUF", "t5xxl-Q4_0.gguf")
64
+ },
65
+ sizeGb: 5.7,
66
+ imageInput: "init",
67
+ defaults: { steps: 28, cfgScale: 4.5, sampler: "euler", size: 1024 }
68
+ },
69
+ {
70
+ id: "flux.1-schnell",
71
+ name: "FLUX.1 schnell",
72
+ files: {
73
+ diffusion_model: hf("second-state/FLUX.1-schnell-GGUF", "flux1-schnell-Q4_0.gguf"),
74
+ vae: FLUX1_VAE,
75
+ clip_l: FLUX1_CLIP_L,
76
+ t5xxl: FLUX1_T5XXL
77
+ },
78
+ sizeGb: 9.9,
79
+ imageInput: "init",
80
+ defaults: { steps: 4, cfgScale: 1, sampler: "euler", size: 1024 }
81
+ },
82
+ {
83
+ id: "flux.1-dev",
84
+ name: "FLUX.1 dev",
85
+ files: {
86
+ diffusion_model: hf("leejet/FLUX.1-dev-gguf", "flux1-dev-q4_0.gguf"),
87
+ vae: FLUX1_VAE,
88
+ clip_l: FLUX1_CLIP_L,
89
+ t5xxl: FLUX1_T5XXL
90
+ },
91
+ sizeGb: 9.9,
92
+ imageInput: "init",
93
+ defaults: { steps: 20, cfgScale: 1, sampler: "euler", size: 1024 }
94
+ },
95
+ {
96
+ id: "flux.1-kontext-dev",
97
+ name: "FLUX.1 Kontext dev",
98
+ files: {
99
+ diffusion_model: hf("QuantStack/FLUX.1-Kontext-dev-GGUF", "flux1-kontext-dev-Q4_0.gguf"),
100
+ vae: FLUX1_VAE,
101
+ clip_l: FLUX1_CLIP_L,
102
+ t5xxl: FLUX1_T5XXL
103
+ },
104
+ sizeGb: 9.9,
105
+ imageInput: "reference",
106
+ defaults: { steps: 20, cfgScale: 1, sampler: "euler", size: 1024 }
107
+ },
108
+ {
109
+ id: "flux.2-dev",
110
+ name: "FLUX.2 dev",
111
+ files: {
112
+ diffusion_model: hf("city96/FLUX.2-dev-gguf", "flux2-dev-Q4_0.gguf"),
113
+ vae: FLUX2_VAE,
114
+ llm: hf("unsloth/Mistral-Small-3.2-24B-Instruct-2506-GGUF", "Mistral-Small-3.2-24B-Instruct-2506-Q4_K_M.gguf")
115
+ },
116
+ sizeGb: 34,
117
+ imageInput: "reference",
118
+ serverArgs: ["--offload-to-cpu"],
119
+ defaults: { steps: 28, cfgScale: 1, sampler: "euler", size: 1024 }
120
+ },
121
+ {
122
+ id: "flux.2-klein-4b",
123
+ name: "FLUX.2 klein 4B",
124
+ files: {
125
+ diffusion_model: hf("leejet/FLUX.2-klein-4B-GGUF", "flux-2-klein-4b-Q4_0.gguf"),
126
+ vae: FLUX2_VAE,
127
+ llm: QWEN3_4B
128
+ },
129
+ sizeGb: 5.3,
130
+ imageInput: "reference",
131
+ defaults: { steps: 4, cfgScale: 1, sampler: "euler", size: 1024 }
132
+ },
133
+ {
134
+ id: "flux.2-klein-base-4b",
135
+ name: "FLUX.2 klein base 4B",
136
+ files: {
137
+ diffusion_model: hf("leejet/FLUX.2-klein-base-4B-GGUF", "flux-2-klein-base-4b-Q4_0.gguf"),
138
+ vae: FLUX2_VAE,
139
+ llm: QWEN3_4B
140
+ },
141
+ sizeGb: 5.3,
142
+ imageInput: "reference",
143
+ defaults: { steps: 20, cfgScale: 4, sampler: "euler", size: 1024 }
144
+ },
145
+ {
146
+ id: "flux.2-klein-9b",
147
+ name: "FLUX.2 klein 9B",
148
+ files: {
149
+ diffusion_model: hf("leejet/FLUX.2-klein-9B-GGUF", "flux-2-klein-9b-Q4_0.gguf"),
150
+ vae: FLUX2_VAE,
151
+ llm: QWEN3_8B
152
+ },
153
+ sizeGb: 11,
154
+ imageInput: "reference",
155
+ defaults: { steps: 4, cfgScale: 1, sampler: "euler", size: 1024 }
156
+ },
157
+ {
158
+ id: "flux.2-klein-base-9b",
159
+ name: "FLUX.2 klein base 9B",
160
+ files: {
161
+ diffusion_model: hf("leejet/FLUX.2-klein-base-9B-GGUF", "flux-2-klein-base-9b-Q4_0.gguf"),
162
+ vae: FLUX2_VAE,
163
+ llm: QWEN3_8B
164
+ },
165
+ sizeGb: 11,
166
+ imageInput: "reference",
167
+ defaults: { steps: 20, cfgScale: 4, sampler: "euler", size: 1024 }
168
+ },
169
+ {
170
+ id: "flux.3-image-klein",
171
+ name: "FLUX.3 Image Klein",
172
+ files: {
173
+ diffusion_model: hf("realrebelai/FLUX-3-Image-Klein_GGUFs", "DISTILLED/flux3_image_klein_sd-Q4_K_M.gguf"),
174
+ vae: FLUX2_VAE,
175
+ llm: QWEN3_VL_4B
176
+ },
177
+ sizeGb: 6.2,
178
+ imageInput: "reference",
179
+ defaults: { steps: 16, cfgScale: 1, sampler: "euler", size: 1024 }
180
+ },
181
+ {
182
+ id: "flux.3-image-klein-base",
183
+ name: "FLUX.3 Image Klein Base",
184
+ files: {
185
+ diffusion_model: hf("realrebelai/FLUX-3-Image-Klein_GGUFs", "BASE/flux3_image_klein-Q4_K_M.gguf"),
186
+ vae: FLUX2_VAE,
187
+ llm: QWEN3_VL_4B
188
+ },
189
+ sizeGb: 6.2,
190
+ imageInput: "reference",
191
+ defaults: { steps: 20, cfgScale: 3.5, sampler: "euler", size: 1024 }
192
+ },
193
+ {
194
+ id: "qwen-image-2.1",
195
+ name: "Qwen Image 2.1",
196
+ files: {
197
+ diffusion_model: hf("leejet/Qwen-Image-2.1-GGUF", "qwen_image_2.1-Q4_K.gguf"),
198
+ vae: hf("Comfy-Org/Qwen-Image-2.1", "vae/qwen_image_2.1_vae_bf16.safetensors"),
199
+ llm: QWEN3_VL_8B,
200
+ llm_vision: QWEN3_VL_8B_VISION
201
+ },
202
+ sizeGb: 11,
203
+ imageInput: "reference",
204
+ serverArgs: ["--offload-to-cpu"],
205
+ defaults: { steps: 40, cfgScale: 6, sampler: "euler", size: 1024 }
206
+ },
207
+ {
208
+ id: "qwen-image-edit-2511",
209
+ name: "Qwen Image Edit 2511",
210
+ files: {
211
+ diffusion_model: hf("unsloth/Qwen-Image-Edit-2511-GGUF", "qwen-image-edit-2511-Q4_0.gguf"),
212
+ vae: QWEN_IMAGE_VAE,
213
+ llm: QWEN25_VL_7B,
214
+ llm_vision: QWEN25_VL_7B_VISION
215
+ },
216
+ sizeGb: 18,
217
+ imageInput: "reference",
218
+ textToImage: false,
219
+ serverArgs: ["--offload-to-cpu", "--model-args", "qwen_image_zero_cond_t=true"],
220
+ defaults: { steps: 40, cfgScale: 2.5, sampler: "euler", flowShift: 3, size: 1024 }
221
+ },
222
+ {
223
+ id: "qwen-image-edit-2509",
224
+ name: "Qwen Image Edit 2509",
225
+ files: {
226
+ diffusion_model: hf("QuantStack/Qwen-Image-Edit-2509-GGUF", "Qwen-Image-Edit-2509-Q4_0.gguf"),
227
+ vae: QWEN_IMAGE_VAE,
228
+ llm: QWEN25_VL_7B,
229
+ llm_vision: QWEN25_VL_7B_VISION
230
+ },
231
+ sizeGb: 18,
232
+ imageInput: "reference",
233
+ textToImage: false,
234
+ serverArgs: ["--offload-to-cpu"],
235
+ defaults: { steps: 40, cfgScale: 2.5, sampler: "euler", flowShift: 3, size: 1024 }
236
+ },
237
+ {
238
+ id: "z-image-turbo",
239
+ name: "Z-Image Turbo",
240
+ files: {
241
+ diffusion_model: hf("leejet/Z-Image-Turbo-GGUF", "z_image_turbo-Q4_K.gguf"),
242
+ vae: FLUX1_VAE,
243
+ llm: QWEN3_4B_2507
244
+ },
245
+ sizeGb: 6.7,
246
+ imageInput: "init",
247
+ defaults: { steps: 8, cfgScale: 1, size: 1024 }
248
+ },
249
+ {
250
+ id: "z-image",
251
+ name: "Z-Image",
252
+ files: {
253
+ diffusion_model: hf("unsloth/Z-Image-GGUF", "z-image-Q4_K_M.gguf"),
254
+ vae: FLUX1_VAE,
255
+ llm: QWEN3_4B_2507
256
+ },
257
+ sizeGb: 7.9,
258
+ imageInput: "init",
259
+ defaults: { steps: 28, cfgScale: 5, size: 1024 }
260
+ },
261
+ {
262
+ id: "z-image-l2p",
263
+ name: "Z-Image L2P",
264
+ files: {
265
+ diffusion_model: hf("zhen-nan/L2P", "model-1k-merge.safetensors"),
266
+ llm: QWEN3_4B_2507
267
+ },
268
+ sizeGb: 22,
269
+ // Pixel space: no VAE, so no source image either.
270
+ serverArgs: ["--type", "bf16"],
271
+ defaults: { steps: 30, cfgScale: 2, sampler: "euler", size: 1024 }
272
+ },
273
+ {
274
+ id: "krea-2-turbo",
275
+ name: "Krea 2 Turbo",
276
+ files: {
277
+ diffusion_model: hf("realrebelai/KREA-2_GGUFs", "TURBO/Krea-2-Turbo-Q4_K_M.gguf"),
278
+ vae: hf("Comfy-Org/Wan_2.1_ComfyUI_repackaged", "split_files/vae/wan_2.1_vae.safetensors"),
279
+ llm: QWEN3_VL_4B
280
+ },
281
+ sizeGb: 10,
282
+ imageInput: "init",
283
+ serverArgs: ["--offload-to-cpu"],
284
+ defaults: { steps: 8, cfgScale: 1, size: 1024 }
285
+ },
286
+ {
287
+ id: "krea-2",
288
+ name: "Krea 2 Raw",
289
+ files: {
290
+ diffusion_model: hf("realrebelai/KREA-2_GGUFs", "BASE/Krea-2-Base-Q4_K_M.gguf"),
291
+ vae: hf("Comfy-Org/Wan_2.1_ComfyUI_repackaged", "split_files/vae/wan_2.1_vae.safetensors"),
292
+ llm: QWEN3_VL_4B
293
+ },
294
+ sizeGb: 10,
295
+ imageInput: "init",
296
+ serverArgs: ["--offload-to-cpu"],
297
+ defaults: { steps: 52, cfgScale: 3.5, size: 1024 }
298
+ },
299
+ {
300
+ id: "ideogram-4",
301
+ name: "Ideogram 4",
302
+ files: {
303
+ diffusion_model: hf("leejet/ideogram-4-GGUF", "ideogram4-Q4_0.gguf"),
304
+ uncond_diffusion_model: hf("leejet/ideogram-4-GGUF", "ideogram4_uncond-Q4_0.gguf"),
305
+ vae: FLUX2_VAE,
306
+ llm: QWEN3_VL_8B
307
+ },
308
+ sizeGb: 17,
309
+ serverArgs: ["--offload-to-cpu"],
310
+ defaults: { steps: 20, cfgScale: 7, size: 1024 }
311
+ },
312
+ {
313
+ id: "iris-3b",
314
+ name: "Iris 3B",
315
+ files: {
316
+ diffusion_model: hf("speridlabs/iris-3b", "model.safetensors"),
317
+ llm: QWEN3_VL_4B
318
+ },
319
+ sizeGb: 14.5,
320
+ // Pixel space: no VAE, so no source image either.
321
+ defaults: { steps: 100, cfgScale: 3, size: 1024 }
322
+ },
323
+ {
324
+ id: "pid-flux-1",
325
+ name: "PiD upscaler (FLUX.1, 512 to 2048)",
326
+ files: {
327
+ diffusion_model: hf("Comfy-Org/PixelDiT", "diffusion_models/pid_flux1_512_to_2048_4step_bf16.safetensors"),
328
+ llm: hf("Comfy-Org/PixelDiT", "text_encoders/gemma_2_2b_it_elm_bf16.safetensors"),
329
+ tokenizer: hf("unsloth/gemma-2-2b", "tokenizer.json"),
330
+ vae: hf("nvidia/PiD", "checkpoints/ae.safetensors")
331
+ },
332
+ sizeGb: 8.3,
333
+ imageInput: "reference",
334
+ textToImage: false,
335
+ serverArgs: ["--vae-format", "flux"],
336
+ sourceScale: { scale: 4, maxSide: 2048 },
337
+ defaults: { steps: 4, cfgScale: 1, size: 2048 }
338
+ },
339
+ {
340
+ id: "anima",
341
+ name: "Anima",
342
+ files: {
343
+ diffusion_model: hf("circlestone-labs/Anima", "split_files/diffusion_models/anima-base-v1.0.safetensors"),
344
+ vae: hf("circlestone-labs/Anima", "split_files/vae/qwen_image_vae.safetensors"),
345
+ llm: hf("circlestone-labs/Anima", "split_files/text_encoders/qwen_3_06b_base.safetensors")
346
+ },
347
+ sizeGb: 5.6,
348
+ imageInput: "init",
349
+ defaults: { steps: 30, cfgScale: 5, sampler: "euler", size: 1024 }
350
+ },
351
+ {
352
+ id: "longcat-image-edit",
353
+ name: "LongCat Image Edit",
354
+ files: {
355
+ diffusion_model: hf("vantagewithai/LongCat-Image-Edit-GGUF", "LongCat-Image-Edit-Q4_K_M.gguf"),
356
+ vae: FLUX1_VAE,
357
+ llm: QWEN25_VL_7B,
358
+ llm_vision: QWEN25_VL_7B_VISION
359
+ },
360
+ sizeGb: 9.8,
361
+ imageInput: "reference",
362
+ textToImage: false,
363
+ serverArgs: ["--offload-to-cpu"],
364
+ defaults: { steps: 50, cfgScale: 4.5, sampler: "euler", flowShift: 3, size: 1024 }
365
+ },
366
+ {
367
+ id: "boogu-image-edit",
368
+ name: "Boogu Image Edit",
369
+ files: {
370
+ diffusion_model: hf("Comfy-Org/Boogu-Image", "diffusion_models/boogu_image_edit_int8_convrot.safetensors"),
371
+ vae: hf("Comfy-Org/Boogu-Image", "vae/flux1_vae_bf16.safetensors"),
372
+ llm: QWEN3_VL_8B,
373
+ llm_vision: QWEN3_VL_8B_VISION
374
+ },
375
+ sizeGb: 17.7,
376
+ imageInput: "reference",
377
+ textToImage: false,
378
+ serverArgs: ["--offload-to-cpu"],
379
+ defaults: { steps: 25, cfgScale: 3.5, size: 1024 }
380
+ },
381
+ {
382
+ id: "mage-flow-edit-turbo",
383
+ name: "Mage-Flow Edit Turbo",
384
+ files: {
385
+ diffusion_model: hf("Comfy-Org/Mage-Flow", "diffusion_models/mage_flow_edit_turbo_int8_convrot.safetensors"),
386
+ vae: hf("Comfy-Org/Mage-Flow", "vae/mage_flow_vae_bf16.safetensors"),
387
+ llm: QWEN3_VL_4B,
388
+ llm_vision: QWEN3_VL_4B_VISION
389
+ },
390
+ sizeGb: 7.9,
391
+ imageInput: "reference",
392
+ textToImage: false,
393
+ defaults: { steps: 4, cfgScale: 1, sampler: "euler", size: 1024 }
394
+ },
395
+ {
396
+ id: "llada-image-turbo",
397
+ name: "LLaDA-Image Turbo",
398
+ files: {
399
+ diffusion_model: hf("fszontagh/LLaDA-Image-Turbo-GGUF", "llada-image-turbo-q8_0.gguf"),
400
+ llm: hf("fszontagh/LLaDA-Image-Turbo-GGUF", "llada-image-turbo-text_encoder-q4_K.gguf"),
401
+ embeddings_connectors: hf("fszontagh/LLaDA-Image-Turbo-GGUF", "llada-image-turbo-connectors-edit.safetensors"),
402
+ tokenizer: hf("inclusionAI/LLaDA-Image-Turbo", "tokenizer/tokenizer.json"),
403
+ vae: hf("inclusionAI/LLaDA-Image-Turbo", "vae/diffusion_pytorch_model.safetensors")
404
+ },
405
+ sizeGb: 20.2,
406
+ imageInput: "reference",
407
+ defaults: { steps: 4, cfgScale: 1, size: 1024 }
408
+ }
409
+ ];
410
+ export function sdCppModel(id) {
411
+ return SD_CPP_MODELS.find((model) => model.id === id) ?? null;
412
+ }
413
+ /** The model's files grouped by repository, in the order the spec lists them. */
414
+ export function sdCppModelRepos(model) {
415
+ const repos = new Map();
416
+ for (const file of Object.values(model.files)) {
417
+ if (!file)
418
+ continue;
419
+ const paths = repos.get(file.repoId) ?? [];
420
+ if (!paths.includes(file.path))
421
+ paths.push(file.path);
422
+ repos.set(file.repoId, paths);
423
+ }
424
+ return [...repos].map(([repoId, paths]) => ({ repoId, paths }));
425
+ }
426
+ /** The provider tasks a model serves. */
427
+ export function sdCppModelTasks(model) {
428
+ const tasks = [];
429
+ if (model.textToImage !== false)
430
+ tasks.push("text_to_image");
431
+ if (model.imageInput)
432
+ tasks.push("image_to_image");
433
+ return tasks;
434
+ }
435
+ //# sourceMappingURL=sdcpp-models.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"sdcpp-models.js","sourceRoot":"","sources":["../../src/providers/sdcpp-models.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAgBH,MAAM,CAAC,MAAM,sBAAsB,GAA6C;IAC9E,KAAK,EAAE,SAAS;IAChB,eAAe,EAAE,mBAAmB;IACpC,sBAAsB,EAAE,0BAA0B;IAClD,GAAG,EAAE,OAAO;IACZ,MAAM,EAAE,UAAU;IAClB,MAAM,EAAE,UAAU;IAClB,KAAK,EAAE,SAAS;IAChB,GAAG,EAAE,OAAO;IACZ,UAAU,EAAE,cAAc;IAC1B,SAAS,EAAE,aAAa;IACxB,qBAAqB,EAAE,yBAAyB;CACjD,CAAC;AA0CF,MAAM,EAAE,GAAG,CAAC,MAAc,EAAE,IAAY,EAAa,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;AAE3E,iCAAiC;AACjC,MAAM,SAAS,GAAG,EAAE,CAAC,kCAAkC,EAAE,gBAAgB,CAAC,CAAC;AAC3E,MAAM,YAAY,GAAG,EAAE,CAAC,kCAAkC,EAAE,kBAAkB,CAAC,CAAC;AAChF,MAAM,WAAW,GAAG,EAAE,CAAC,kCAAkC,EAAE,iBAAiB,CAAC,CAAC;AAC9E,MAAM,SAAS,GAAG,EAAE,CAAC,qBAAqB,EAAE,uCAAuC,CAAC,CAAC;AACrF,MAAM,cAAc,GAAG,EAAE,CAAC,8BAA8B,EAAE,4CAA4C,CAAC,CAAC;AACxG,MAAM,QAAQ,GAAG,EAAE,CAAC,uBAAuB,EAAE,sBAAsB,CAAC,CAAC;AACrE,MAAM,QAAQ,GAAG,EAAE,CAAC,uBAAuB,EAAE,sBAAsB,CAAC,CAAC;AACrE,wDAAwD;AACxD,MAAM,aAAa,GAAG,EAAE,CAAC,qCAAqC,EAAE,oCAAoC,CAAC,CAAC;AACtG,MAAM,WAAW,GAAG,EAAE,CAAC,gCAAgC,EAAE,iCAAiC,CAAC,CAAC;AAC5F,MAAM,kBAAkB,GAAG,EAAE,CAAC,gCAAgC,EAAE,qCAAqC,CAAC,CAAC;AACvG,MAAM,WAAW,GAAG,EAAE,CAAC,gCAAgC,EAAE,iCAAiC,CAAC,CAAC;AAC5F,MAAM,kBAAkB,GAAG,EAAE,CAAC,gCAAgC,EAAE,qCAAqC,CAAC,CAAC;AACvG,MAAM,YAAY,GAAG,EAAE,CAAC,0CAA0C,EAAE,oCAAoC,CAAC,CAAC;AAC1G,MAAM,mBAAmB,GAAG,EAAE,CAAC,0CAA0C,EAAE,yCAAyC,CAAC,CAAC;AAEtH,MAAM,CAAC,MAAM,aAAa,GAA8B;IACtD;QACE,EAAE,EAAE,UAAU;QACd,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,yBAAyB,EAAE,wBAAwB,CAAC,EAAE;QACzE,MAAM,EAAE,CAAC;QACT,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE;KAC/C;IACD;QACE,EAAE,EAAE,QAAQ;QACZ,IAAI,EAAE,sBAAsB;QAC5B,KAAK,EAAE;YACL,KAAK,EAAE,EAAE,CAAC,yCAAyC,EAAE,gDAAgD,CAAC;SACvG;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE;KAChD;IACD;QACE,EAAE,EAAE,eAAe;QACnB,IAAI,EAAE,6BAA6B;QACnC,KAAK,EAAE;YACL,KAAK,EAAE,EAAE,CAAC,+CAA+C,EAAE,wBAAwB,CAAC;YACpF,MAAM,EAAE,EAAE,CAAC,+CAA+C,EAAE,kBAAkB,CAAC;YAC/E,MAAM,EAAE,EAAE,CAAC,+CAA+C,EAAE,kBAAkB,CAAC;YAC/E,KAAK,EAAE,EAAE,CAAC,+CAA+C,EAAE,iBAAiB,CAAC;SAC9E;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACrE;IACD;QACE,EAAE,EAAE,gBAAgB;QACpB,IAAI,EAAE,gBAAgB;QACtB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,kCAAkC,EAAE,yBAAyB,CAAC;YAClF,GAAG,EAAE,SAAS;YACd,MAAM,EAAE,YAAY;YACpB,KAAK,EAAE,WAAW;SACnB;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KAClE;IACD;QACE,EAAE,EAAE,YAAY;QAChB,IAAI,EAAE,YAAY;QAClB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,wBAAwB,EAAE,qBAAqB,CAAC;YACpE,GAAG,EAAE,SAAS;YACd,MAAM,EAAE,YAAY;YACpB,KAAK,EAAE,WAAW;SACnB;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACnE;IACD;QACE,EAAE,EAAE,oBAAoB;QACxB,IAAI,EAAE,oBAAoB;QAC1B,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,oCAAoC,EAAE,6BAA6B,CAAC;YACxF,GAAG,EAAE,SAAS;YACd,MAAM,EAAE,YAAY;YACpB,KAAK,EAAE,WAAW;SACnB;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,WAAW;QACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACnE;IACD;QACE,EAAE,EAAE,YAAY;QAChB,IAAI,EAAE,YAAY;QAClB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,wBAAwB,EAAE,qBAAqB,CAAC;YACpE,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,EAAE,CAAC,kDAAkD,EAAE,iDAAiD,CAAC;SAC/G;QACD,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,WAAW;QACvB,UAAU,EAAE,CAAC,kBAAkB,CAAC;QAChC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACnE;IACD;QACE,EAAE,EAAE,iBAAiB;QACrB,IAAI,EAAE,iBAAiB;QACvB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,6BAA6B,EAAE,2BAA2B,CAAC;YAC/E,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,QAAQ;SACd;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,WAAW;QACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KAClE;IACD;QACE,EAAE,EAAE,sBAAsB;QAC1B,IAAI,EAAE,sBAAsB;QAC5B,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,kCAAkC,EAAE,gCAAgC,CAAC;YACzF,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,QAAQ;SACd;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,WAAW;QACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACnE;IACD;QACE,EAAE,EAAE,iBAAiB;QACrB,IAAI,EAAE,iBAAiB;QACvB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,6BAA6B,EAAE,2BAA2B,CAAC;YAC/E,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,QAAQ;SACd;QACD,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,WAAW;QACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KAClE;IACD;QACE,EAAE,EAAE,sBAAsB;QAC1B,IAAI,EAAE,sBAAsB;QAC5B,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,kCAAkC,EAAE,gCAAgC,CAAC;YACzF,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,QAAQ;SACd;QACD,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,WAAW;QACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACnE;IACD;QACE,EAAE,EAAE,oBAAoB;QACxB,IAAI,EAAE,oBAAoB;QAC1B,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,sCAAsC,EAAE,4CAA4C,CAAC;YACzG,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,WAAW;SACjB;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,WAAW;QACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACnE;IACD;QACE,EAAE,EAAE,yBAAyB;QAC7B,IAAI,EAAE,yBAAyB;QAC/B,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,sCAAsC,EAAE,oCAAoC,CAAC;YACjG,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,WAAW;SACjB;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,WAAW;QACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACrE;IACD;QACE,EAAE,EAAE,gBAAgB;QACpB,IAAI,EAAE,gBAAgB;QACtB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,4BAA4B,EAAE,0BAA0B,CAAC;YAC7E,GAAG,EAAE,EAAE,CAAC,0BAA0B,EAAE,yCAAyC,CAAC;YAC9E,GAAG,EAAE,WAAW;YAChB,UAAU,EAAE,kBAAkB;SAC/B;QACD,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,WAAW;QACvB,UAAU,EAAE,CAAC,kBAAkB,CAAC;QAChC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACnE;IACD;QACE,EAAE,EAAE,sBAAsB;QAC1B,IAAI,EAAE,sBAAsB;QAC5B,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,mCAAmC,EAAE,gCAAgC,CAAC;YAC1F,GAAG,EAAE,cAAc;YACnB,GAAG,EAAE,YAAY;YACjB,UAAU,EAAE,mBAAmB;SAChC;QACD,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,WAAW;QACvB,WAAW,EAAE,KAAK;QAClB,UAAU,EAAE,CAAC,kBAAkB,EAAE,cAAc,EAAE,6BAA6B,CAAC;QAC/E,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KACnF;IACD;QACE,EAAE,EAAE,sBAAsB;QAC1B,IAAI,EAAE,sBAAsB;QAC5B,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,sCAAsC,EAAE,gCAAgC,CAAC;YAC7F,GAAG,EAAE,cAAc;YACnB,GAAG,EAAE,YAAY;YACjB,UAAU,EAAE,mBAAmB;SAChC;QACD,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,WAAW;QACvB,WAAW,EAAE,KAAK;QAClB,UAAU,EAAE,CAAC,kBAAkB,CAAC;QAChC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KACnF;IACD;QACE,EAAE,EAAE,eAAe;QACnB,IAAI,EAAE,eAAe;QACrB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,2BAA2B,EAAE,yBAAyB,CAAC;YAC3E,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,aAAa;SACnB;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KAChD;IACD;QACE,EAAE,EAAE,SAAS;QACb,IAAI,EAAE,SAAS;QACf,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,sBAAsB,EAAE,qBAAqB,CAAC;YAClE,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,aAAa;SACnB;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KACjD;IACD;QACE,EAAE,EAAE,aAAa;QACjB,IAAI,EAAE,aAAa;QACnB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,cAAc,EAAE,4BAA4B,CAAC;YACjE,GAAG,EAAE,aAAa;SACnB;QACD,MAAM,EAAE,EAAE;QACV,kDAAkD;QAClD,UAAU,EAAE,CAAC,QAAQ,EAAE,MAAM,CAAC;QAC9B,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACnE;IACD;QACE,EAAE,EAAE,cAAc;QAClB,IAAI,EAAE,cAAc;QACpB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,0BAA0B,EAAE,gCAAgC,CAAC;YACjF,GAAG,EAAE,EAAE,CAAC,sCAAsC,EAAE,yCAAyC,CAAC;YAC1F,GAAG,EAAE,WAAW;SACjB;QACD,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,MAAM;QAClB,UAAU,EAAE,CAAC,kBAAkB,CAAC;QAChC,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KAChD;IACD;QACE,EAAE,EAAE,QAAQ;QACZ,IAAI,EAAE,YAAY;QAClB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,0BAA0B,EAAE,8BAA8B,CAAC;YAC/E,GAAG,EAAE,EAAE,CAAC,sCAAsC,EAAE,yCAAyC,CAAC;YAC1F,GAAG,EAAE,WAAW;SACjB;QACD,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,MAAM;QAClB,UAAU,EAAE,CAAC,kBAAkB,CAAC;QAChC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE;KACnD;IACD;QACE,EAAE,EAAE,YAAY;QAChB,IAAI,EAAE,YAAY;QAClB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,wBAAwB,EAAE,qBAAqB,CAAC;YACpE,sBAAsB,EAAE,EAAE,CAAC,wBAAwB,EAAE,4BAA4B,CAAC;YAClF,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,WAAW;SACjB;QACD,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,CAAC,kBAAkB,CAAC;QAChC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KACjD;IACD;QACE,EAAE,EAAE,SAAS;QACb,IAAI,EAAE,SAAS;QACf,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,oBAAoB,EAAE,mBAAmB,CAAC;YAC9D,GAAG,EAAE,WAAW;SACjB;QACD,MAAM,EAAE,IAAI;QACZ,kDAAkD;QAClD,QAAQ,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KAClD;IACD;QACE,EAAE,EAAE,YAAY;QAChB,IAAI,EAAE,oCAAoC;QAC1C,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,oBAAoB,EAAE,+DAA+D,CAAC;YAC1G,GAAG,EAAE,EAAE,CAAC,oBAAoB,EAAE,kDAAkD,CAAC;YACjF,SAAS,EAAE,EAAE,CAAC,oBAAoB,EAAE,gBAAgB,CAAC;YACrD,GAAG,EAAE,EAAE,CAAC,YAAY,EAAE,4BAA4B,CAAC;SACpD;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,WAAW;QACvB,WAAW,EAAE,KAAK;QAClB,UAAU,EAAE,CAAC,cAAc,EAAE,MAAM,CAAC;QACpC,WAAW,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE;QACxC,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KAChD;IACD;QACE,EAAE,EAAE,OAAO;QACX,IAAI,EAAE,OAAO;QACb,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,wBAAwB,EAAE,0DAA0D,CAAC;YACzG,GAAG,EAAE,EAAE,CAAC,wBAAwB,EAAE,4CAA4C,CAAC;YAC/E,GAAG,EAAE,EAAE,CAAC,wBAAwB,EAAE,uDAAuD,CAAC;SAC3F;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,MAAM;QAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KACnE;IACD;QACE,EAAE,EAAE,oBAAoB;QACxB,IAAI,EAAE,oBAAoB;QAC1B,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,uCAAuC,EAAE,gCAAgC,CAAC;YAC9F,GAAG,EAAE,SAAS;YACd,GAAG,EAAE,YAAY;YACjB,UAAU,EAAE,mBAAmB;SAChC;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,WAAW;QACvB,WAAW,EAAE,KAAK;QAClB,UAAU,EAAE,CAAC,kBAAkB,CAAC;QAChC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KACnF;IACD;QACE,EAAE,EAAE,kBAAkB;QACtB,IAAI,EAAE,kBAAkB;QACxB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,uBAAuB,EAAE,4DAA4D,CAAC;YAC1G,GAAG,EAAE,EAAE,CAAC,uBAAuB,EAAE,gCAAgC,CAAC;YAClE,GAAG,EAAE,WAAW;YAChB,UAAU,EAAE,kBAAkB;SAC/B;QACD,MAAM,EAAE,IAAI;QACZ,UAAU,EAAE,WAAW;QACvB,WAAW,EAAE,KAAK;QAClB,UAAU,EAAE,CAAC,kBAAkB,CAAC;QAChC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE;KACnD;IACD;QACE,EAAE,EAAE,sBAAsB;QAC1B,IAAI,EAAE,sBAAsB;QAC5B,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,qBAAqB,EAAE,gEAAgE,CAAC;YAC5G,GAAG,EAAE,EAAE,CAAC,qBAAqB,EAAE,oCAAoC,CAAC;YACpE,GAAG,EAAE,WAAW;YAChB,UAAU,EAAE,kBAAkB;SAC/B;QACD,MAAM,EAAE,GAAG;QACX,UAAU,EAAE,WAAW;QACvB,WAAW,EAAE,KAAK;QAClB,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;KAClE;IACD;QACE,EAAE,EAAE,mBAAmB;QACvB,IAAI,EAAE,mBAAmB;QACzB,KAAK,EAAE;YACL,eAAe,EAAE,EAAE,CAAC,kCAAkC,EAAE,6BAA6B,CAAC;YACtF,GAAG,EAAE,EAAE,CAAC,kCAAkC,EAAE,0CAA0C,CAAC;YACvF,qBAAqB,EAAE,EAAE,CAAC,kCAAkC,EAAE,+CAA+C,CAAC;YAC9G,SAAS,EAAE,EAAE,CAAC,+BAA+B,EAAE,0BAA0B,CAAC;YAC1E,GAAG,EAAE,EAAE,CAAC,+BAA+B,EAAE,yCAAyC,CAAC;SACpF;QACD,MAAM,EAAE,IAAI;QACZ,UAAU,EAAE,WAAW;QACvB,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE;KAChD;CACF,CAAC;AAEF,MAAM,UAAU,UAAU,CAAC,EAAU;IACnC,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,IAAI,CAAC;AAChE,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,eAAe,CAAC,KAAqB;IACnD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC1C,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;QAC9C,IAAI,CAAC,IAAI;YAAE,SAAS;QACpB,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC3C,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;YAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAChC,CAAC;IACD,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;AAClE,CAAC;AAED,yCAAyC;AACzC,MAAM,UAAU,eAAe,CAAC,KAAqB;IACnD,MAAM,KAAK,GAA2C,EAAE,CAAC;IACzD,IAAI,KAAK,CAAC,WAAW,KAAK,KAAK;QAAE,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC7D,IAAI,KAAK,CAAC,UAAU;QAAE,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACnD,OAAO,KAAK,CAAC;AACf,CAAC"}
@@ -0,0 +1,40 @@
1
+ import { BaseProvider, type ProviderCapability } from "./base-provider.js";
2
+ import type { ImageModel, ImageToImageParams, Message, ProviderStreamItem, TextToImageParams } from "./types.js";
3
+ import { type SdCppModelSpec } from "./sdcpp-models.js";
4
+ export declare const SD_CPP_PROVIDER_ID = "stable_diffusion_cpp";
5
+ export declare class StableDiffusionCppProvider extends BaseProvider {
6
+ static requiredSecrets(): string[];
7
+ private readonly serverPath?;
8
+ constructor(secrets?: {
9
+ SD_CPP_SERVER_PATH?: string;
10
+ });
11
+ protected declaredCapabilities(): readonly ProviderCapability[];
12
+ generateMessage(_args: Parameters<BaseProvider["generateMessage"]>[0]): Promise<Message>;
13
+ generateMessages(_args: Parameters<BaseProvider["generateMessages"]>[0]): AsyncGenerator<ProviderStreamItem>;
14
+ getAvailableImageModels(): Promise<ImageModel[]>;
15
+ textToImage(params: TextToImageParams): Promise<Uint8Array>;
16
+ imageToImage(images: Uint8Array[], params: ImageToImageParams): Promise<Uint8Array>;
17
+ private spec;
18
+ private generate;
19
+ /**
20
+ * The `sd-server` to run: the configured path, the desktop or CLI install,
21
+ * then PATH. Null when none exists.
22
+ */
23
+ private findServer;
24
+ }
25
+ /**
26
+ * The output size: the requested width and height, else the aspect ratio at
27
+ * the model's trained area, else a square at the trained size. With `fit`
28
+ * (a size read from a source image), a size above 1.5 times the trained area
29
+ * scales down to it, keeping the aspect ratio.
30
+ */
31
+ export declare function outputSize(model: SdCppModelSpec, width?: number | null, height?: number | null, aspectRatio?: string | null, fit?: boolean): {
32
+ width: number;
33
+ height: number;
34
+ };
35
+ /** Width and height from a PNG or JPEG header, or null for other formats. */
36
+ export declare function imageDimensions(bytes: Uint8Array): {
37
+ width: number;
38
+ height: number;
39
+ } | null;
40
+ //# sourceMappingURL=sdcpp-provider.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"sdcpp-provider.d.ts","sourceRoot":"","sources":["../../src/providers/sdcpp-provider.ts"],"names":[],"mappings":"AAiBA,OAAO,EAAE,YAAY,EAAE,KAAK,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAC3E,OAAO,KAAK,EACV,UAAU,EACV,kBAAkB,EAClB,OAAO,EAEP,kBAAkB,EAClB,iBAAiB,EAClB,MAAM,YAAY,CAAC;AACpB,OAAO,EAQL,KAAK,cAAc,EACpB,MAAM,mBAAmB,CAAC;AAI3B,eAAO,MAAM,kBAAkB,yBAAyB,CAAC;AAezD,qBAAa,0BAA2B,SAAQ,YAAY;IAC1D,MAAM,CAAC,eAAe,IAAI,MAAM,EAAE,CAEjC;IAED,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAS;IAErC,YAAY,OAAO,GAAE;QAAE,kBAAkB,CAAC,EAAE,MAAM,CAAA;KAAO,EAMxD;IAED,UAAmB,oBAAoB,IAAI,SAAS,kBAAkB,EAAE,CAEvE;IAEK,eAAe,CACnB,KAAK,EAAE,UAAU,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC,GACpD,OAAO,CAAC,OAAO,CAAC,CAElB;IAGM,gBAAgB,CACrB,KAAK,EAAE,UAAU,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC,GACrD,cAAc,CAAC,kBAAkB,CAAC,CAEpC;IAEc,uBAAuB,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC,CAW9D;IAEc,WAAW,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,UAAU,CAAC,CAczE;IAEc,YAAY,CACzB,MAAM,EAAE,UAAU,EAAE,EACpB,MAAM,EAAE,kBAAkB,GACzB,OAAO,CAAC,UAAU,CAAC,CAkCrB;IAED,OAAO,CAAC,IAAI;YAUE,QAAQ;IAoBtB;;;OAGG;YACW,UAAU;CAsCzB;AAqHD;;;;;GAKG;AACH,wBAAgB,UAAU,CACxB,KAAK,EAAE,cAAc,EACrB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,EACrB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,EACtB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,EAC3B,GAAG,UAAQ,GACV;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,CAWnC;AAWD,6EAA6E;AAC7E,wBAAgB,eAAe,CAAC,KAAK,EAAE,UAAU,GAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAoB3F"}