mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-03 02:57:05 +08:00
test
This commit is contained in:
parent
ee8a636cb7
commit
075e17041b
@ -440,7 +440,7 @@ class WanModel(torch.nn.Module):
|
|||||||
|
|
||||||
# embeddings
|
# embeddings
|
||||||
self.patch_embedding = operations.Conv3d(
|
self.patch_embedding = operations.Conv3d(
|
||||||
20, dim, kernel_size=patch_size, stride=patch_size, device=operation_settings.get("device"), dtype=torch.float32)
|
in_dim, dim, kernel_size=patch_size, stride=patch_size, device=operation_settings.get("device"), dtype=torch.float32)
|
||||||
self.text_embedding = nn.Sequential(
|
self.text_embedding = nn.Sequential(
|
||||||
operations.Linear(text_dim, dim, device=operation_settings.get("device"), dtype=operation_settings.get("dtype")), nn.GELU(approximate='tanh'),
|
operations.Linear(text_dim, dim, device=operation_settings.get("device"), dtype=operation_settings.get("dtype")), nn.GELU(approximate='tanh'),
|
||||||
operations.Linear(dim, dim, device=operation_settings.get("device"), dtype=operation_settings.get("dtype")))
|
operations.Linear(dim, dim, device=operation_settings.get("device"), dtype=operation_settings.get("dtype")))
|
||||||
|
|||||||
@ -672,11 +672,11 @@ class WanTrackToVideo:
|
|||||||
mask, video = motion_patched[:, 0:4], motion_patched[:, 4:]
|
mask, video = motion_patched[:, 0:4], motion_patched[:, 4:]
|
||||||
# Add motion features to conditioning
|
# Add motion features to conditioning
|
||||||
positive = node_helpers.conditioning_set_values(positive,
|
positive = node_helpers.conditioning_set_values(positive,
|
||||||
{"concat_mask": mask,
|
{"concat_mask": motion_patched,
|
||||||
"concat_latent_image": video})
|
"concat_latent_image": image})
|
||||||
negative = node_helpers.conditioning_set_values(negative,
|
negative = node_helpers.conditioning_set_values(negative,
|
||||||
{"concat_mask": mask,
|
{"concat_mask": motion_patched,
|
||||||
"concat_latent_image": video})
|
"concat_latent_image": image})
|
||||||
|
|
||||||
|
|
||||||
# Handle clip vision output if provided
|
# Handle clip vision output if provided
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user