This commit is contained in:
Eugene Fairley 2025-06-27 11:51:55 -07:00
parent ee8a636cb7
commit 075e17041b
2 changed files with 5 additions and 5 deletions

View File

@ -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")))

View File

@ -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