mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-11-11 11:26:34 +00:00
feat: add P-H3-Image generation and editing support (#2272)
This commit is contained in:
1 parent
984eb0686c
commit
3e92bf990a
18 files changed
+421
-41
No files matched your search
@@ -206,7 +206,7 @@ namespace sd {
|
||||
for (size_t seg_idx = 0; seg_idx < plan.segments.size(); seg_idx++) {
|
||||
ggml_backend_t backend = backend_by_segment[seg_idx];
|
||||
const auto& segment = plan.segments[seg_idx];
|
||||
auto assign_node = [&](int node_index) {
|
||||
auto assign_node = [&](int node_index) {
|
||||
if (node_index < 0 || node_index >= n_nodes) {
|
||||
return;
|
||||
}
|
||||
|
||||
@@ -1,16 +1,25 @@
|
||||
#version 450
|
||||
|
||||
layout(local_size_x = 256) in;
|
||||
layout(binding = 0) readonly buffer X { float x[]; };
|
||||
layout(binding = 1) readonly buffer S { float scale[]; };
|
||||
layout(binding = 2) readonly buffer H { float shift[]; };
|
||||
layout(binding = 3) writeonly buffer D { float dst[]; };
|
||||
layout(binding = 0) readonly buffer X {
|
||||
float x[];
|
||||
};
|
||||
layout(binding = 1) readonly buffer S {
|
||||
float scale[];
|
||||
};
|
||||
layout(binding = 2) readonly buffer H {
|
||||
float shift[];
|
||||
};
|
||||
layout(binding = 3) writeonly buffer D {
|
||||
float dst[];
|
||||
};
|
||||
layout(push_constant) uniform Parameters {
|
||||
uint ne[4], scale_ne[4], shift_ne[4];
|
||||
uint x_stride[4], scale_stride[4], shift_stride[4];
|
||||
uint offsets[4];
|
||||
uint count, has_shift;
|
||||
} p;
|
||||
}
|
||||
p;
|
||||
|
||||
void main() {
|
||||
uint group = gl_WorkGroupID.y * gl_NumWorkGroups.x + gl_WorkGroupID.x;
|
||||
|
||||
@@ -2,8 +2,8 @@
|
||||
|
||||
#extension GL_EXT_control_flow_attributes : require
|
||||
|
||||
layout(constant_id = 0) const uint K = 4096;
|
||||
layout(constant_id = 1) const bool AFFINE = false;
|
||||
layout(constant_id = 0) const uint K = 4096;
|
||||
layout(constant_id = 1) const bool AFFINE = false;
|
||||
layout(constant_id = 2) const bool HAS_SHIFT = true;
|
||||
layout(local_size_x = 512, local_size_y = 1, local_size_z = 1) in;
|
||||
layout(binding = 0) readonly buffer X {
|
||||
|
||||
Reference in new issue
Block a user