xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-15 07:54:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-08-28 00:56:15 +03:00
parent 4a75855740
commit 58fee8e567
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes.py
View File

@ -71,11 +71,11 @@ class DownloadAndLoadCogVideoModel:
pipe = CogVideoXPipeline(vae, transformer, scheduler) pipe = CogVideoXPipeline(vae, transformer, scheduler)
if torch_compile: if torch_compile:
torch._dynamo.config.suppress_errors = True
torch._inductor.config.conv_1x1_as_mm = True torch._inductor.config.conv_1x1_as_mm = True
torch._inductor.config.coordinate_descent_tuning = True torch._inductor.config.coordinate_descent_tuning = True
torch._inductor.config.epilogue_fusion = False torch._inductor.config.epilogue_fusion = False
torch._inductor.config.coordinate_descent_check_all_directions = True torch._inductor.config.coordinate_descent_check_all_directions = True
torch._inductor.config.num_stages = 1
pipe.transformer.to(memory_format=torch.channels_last) pipe.transformer.to(memory_format=torch.channels_last)
pipe.transformer = torch.compile(pipe.transformer, mode="max-autotune", fullgraph=True) pipe.transformer = torch.compile(pipe.transformer, mode="max-autotune", fullgraph=True)