mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2025-12-13 05:48:56 +00:00
fix get_first_stage_encoding
This commit is contained in:
parent
a8d3aa0415
commit
a3a2b2d721
@ -1510,7 +1510,7 @@ public:
|
|||||||
|
|
||||||
ggml_tensor* get_first_stage_encoding(ggml_context* work_ctx, ggml_tensor* vae_output) {
|
ggml_tensor* get_first_stage_encoding(ggml_context* work_ctx, ggml_tensor* vae_output) {
|
||||||
ggml_tensor* latent;
|
ggml_tensor* latent;
|
||||||
if (use_tiny_autoencoder || sd_version_is_qwen_image(version)) {
|
if (use_tiny_autoencoder || sd_version_is_qwen_image(version) || sd_version_is_wan(version)) {
|
||||||
latent = vae_output;
|
latent = vae_output;
|
||||||
} else if (version == VERSION_SD1_PIX2PIX) {
|
} else if (version == VERSION_SD1_PIX2PIX) {
|
||||||
latent = ggml_view_3d(work_ctx,
|
latent = ggml_view_3d(work_ctx,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user