mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-11-14 12:56:33 +00:00
fix get_first_stage_encoding
This commit is contained in:
1 parent
a8d3aa0415
commit
a3a2b2d721
1 file changed
+1
-1
@@ -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,
|
||||||
|
|||||||
Reference in new issue
Block a user