mirror of
https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git
synced 2025-12-10 05:14:22 +08:00
Update nodes.py
This commit is contained in:
parent
cb5465ad20
commit
4b7fdac56d
4
nodes.py
4
nodes.py
@ -72,10 +72,6 @@ class DownloadAndLoadCogVideoModel:
|
|||||||
|
|
||||||
if torch_compile:
|
if torch_compile:
|
||||||
torch._dynamo.config.suppress_errors = True
|
torch._dynamo.config.suppress_errors = True
|
||||||
torch._inductor.config.conv_1x1_as_mm = True
|
|
||||||
torch._inductor.config.coordinate_descent_tuning = True
|
|
||||||
torch._inductor.config.epilogue_fusion = False
|
|
||||||
torch._inductor.config.coordinate_descent_check_all_directions = True
|
|
||||||
pipe.transformer.to(device).to(memory_format=torch.channels_last)
|
pipe.transformer.to(device).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)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user