mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-02 23:47:18 +08:00
test
This commit is contained in:
parent
ee8a636cb7
commit
075e17041b
@ -440,7 +440,7 @@ class WanModel(torch.nn.Module):
|
||||
|
||||
# embeddings
|
||||
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(
|
||||
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")))
|
||||
|
||||
@ -672,11 +672,11 @@ class WanTrackToVideo:
|
||||
mask, video = motion_patched[:, 0:4], motion_patched[:, 4:]
|
||||
# Add motion features to conditioning
|
||||
positive = node_helpers.conditioning_set_values(positive,
|
||||
{"concat_mask": mask,
|
||||
"concat_latent_image": video})
|
||||
{"concat_mask": motion_patched,
|
||||
"concat_latent_image": image})
|
||||
negative = node_helpers.conditioning_set_values(negative,
|
||||
{"concat_mask": mask,
|
||||
"concat_latent_image": video})
|
||||
{"concat_mask": motion_patched,
|
||||
"concat_latent_image": image})
|
||||
|
||||
|
||||
# Handle clip vision output if provided
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user