wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 23:00:51 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

fix linting issues

Browse Source
This commit is contained in:
doctorpangloss 2025-12-12 12:29:26 -08:00
parent ff66341112
commit 520c5f66be
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes/nodes_wanmove.py
View File

@ -1,4 +1,4 @@
import nodes from comfy.nodes import base_nodes as nodes
from comfy import node_helpers from comfy import node_helpers
import torch import torch
import torchvision.transforms.functional as TF import torchvision.transforms.functional as TF
@ -7,7 +7,7 @@ import comfy.utils
import numpy as np import numpy as np
from typing_extensions import override from typing_extensions import override
from comfy_api.latest import ComfyExtension, io from comfy_api.latest import ComfyExtension, io
from comfy_extras.nodes.nodes_wan import parse_json_tracks from .nodes_wan import parse_json_tracks
# https://github.com/ali-vilab/Wan-Move/blob/main/wan/modules/trajectory.py # https://github.com/ali-vilab/Wan-Move/blob/main/wan/modules/trajectory.py
from PIL import Image, ImageDraw from PIL import Image, ImageDraw
@ -372,6 +372,7 @@ class GenerateTracks(io.ComfyNode):
track_spread_px = track_spread * (width + height) / 2 # Use average of width/height for spread to keep it proportional track_spread_px = track_spread * (width + height) / 2 # Use average of width/height for spread to keep it proportional
t = torch.linspace(0, 1, num_frames, device=device) t = torch.linspace(0, 1, num_frames, device=device)
interp_values = None
if interpolation == "constant": # All points stay at start position if interpolation == "constant": # All points stay at start position
interp_values = torch.zeros_like(t) interp_values = torch.zeros_like(t)
elif interpolation == "linear": elif interpolation == "linear":