xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-09 21:04:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added validation for max_size parameter in get_3d_rotary_pos_embed function when grid_type is set to 'slice'.

Browse Source
This commit is contained in:
sko00o 2025-07-31 15:01:48 +08:00
parent 389fb0323f
commit 881bbbf6c9
No known key found for this signature in database
GPG Key ID: 240DEE2151D0CF19
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
embeddings.py
View File

@ -174,6 +174,8 @@ def get_3d_rotary_pos_embed(
grid_t = np.arange(temporal_size, dtype=np.float32) grid_t = np.arange(temporal_size, dtype=np.float32)
grid_t = np.linspace(0, temporal_size, temporal_size, endpoint=False, dtype=np.float32) grid_t = np.linspace(0, temporal_size, temporal_size, endpoint=False, dtype=np.float32)
elif grid_type == "slice": elif grid_type == "slice":
if max_size is None:
raise ValueError("`max_size` must be provided when `grid_type` is 'slice'")
max_h, max_w = max_size max_h, max_w = max_size
grid_size_h, grid_size_w = grid_size grid_size_h, grid_size_w = grid_size
grid_h = np.arange(max_h, dtype=np.float32) grid_h = np.arange(max_h, dtype=np.float32)