xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-02 13:00:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove print

Browse Source
This commit is contained in:
kijai 2024-10-26 02:35:29 +03:00
parent dcca095743
commit c17750ea0a
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
nodes.py
View File

@ -411,7 +411,6 @@ class DownloadAndLoadCogVideoModel:
#pipe.transformer = torch.compile(pipe.transformer, mode="default", fullgraph=False, backend="inductor") #pipe.transformer = torch.compile(pipe.transformer, mode="default", fullgraph=False, backend="inductor")
for i, block in enumerate(pipe.transformer.transformer_blocks): for i, block in enumerate(pipe.transformer.transformer_blocks):
if "CogVideoXBlock" in str(block): if "CogVideoXBlock" in str(block):
print(block)
pipe.transformer.transformer_blocks[i] = torch.compile(block, fullgraph=False, dynamic=False, backend="inductor") pipe.transformer.transformer_blocks[i] = torch.compile(block, fullgraph=False, dynamic=False, backend="inductor")
elif compile == "onediff": elif compile == "onediff":
from onediffx import compile_pipe from onediffx import compile_pipe