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

LTX2 context windows - Move symmetric_patchifier import to module level

Browse Source
This commit is contained in:
ozbayb 2026-04-12 15:46:20 -06:00
parent d59d6fb7a0
commit f72583d1f3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_base.py
View File

@ -21,6 +21,7 @@ import comfy.ldm.hunyuan3dv2_1.hunyuandit
import torch import torch
import logging import logging
import comfy.ldm.lightricks.av_model import comfy.ldm.lightricks.av_model
import comfy.ldm.lightricks.symmetric_patchifier
import comfy.context_windows import comfy.context_windows
from comfy.ldm.modules.diffusionmodules.openaimodel import UNetModel, Timestep from comfy.ldm.modules.diffusionmodules.openaimodel import UNetModel, Timestep
from comfy.ldm.cascade.stage_c import StageC from comfy.ldm.cascade.stage_c import StageC
@ -1156,9 +1157,8 @@ class LTXAV(BaseModel):
window_len = len(window.index_list) window_len = len(window.index_list)
patchifier = self.diffusion_model.patchifier patchifier = self.diffusion_model.patchifier
latent_coords = patchifier.get_latent_coords(window_len, H, W, 1, cond_value.cond.device) latent_coords = patchifier.get_latent_coords(window_len, H, W, 1, cond_value.cond.device)
from comfy.ldm.lightricks.symmetric_patchifier import latent_to_pixel_coords
scale_factors = self.diffusion_model.vae_scale_factors scale_factors = self.diffusion_model.vae_scale_factors
pixel_coords = latent_to_pixel_coords( pixel_coords = comfy.ldm.lightricks.symmetric_patchifier.latent_to_pixel_coords(
latent_coords, latent_coords,
scale_factors, scale_factors,
causal_fix=self.diffusion_model.causal_temporal_positioning) causal_fix=self.diffusion_model.causal_temporal_positioning)