mirror of
https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git
synced 2025-12-09 04:44: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:
|
||||
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 = torch.compile(pipe.transformer, mode="max-autotune", fullgraph=True)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user