Fix dimension ordering in HuMo reference images processing
This commit is contained in:
+2
-2
@@ -209,7 +209,7 @@ class HuMoEmbeds:
|
||||
num_refs = 0
|
||||
if reference_images is not None:
|
||||
if reference_images.shape[1] != height or reference_images.shape[2] != width:
|
||||
reference_images_in = common_upscale(reference_images.movedim(-1, 1), width, height, "lanczos", "disabled").movedim(-1, 1)
|
||||
reference_images_in = common_upscale(reference_images.movedim(-1, 1), width, height, "lanczos", "disabled").movedim(1, -1)
|
||||
else:
|
||||
reference_images_in = reference_images
|
||||
samples, = WanVideoEncodeLatentBatch.encode(self, vae, reference_images_in, tiled_vae, None, None, None, None)
|
||||
@@ -262,4 +262,4 @@ NODE_CLASS_MAPPINGS = {
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
"WhisperModelLoader": "Whisper Model Loader",
|
||||
"HuMoEmbeds": "HuMo Embeds",
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user