mirror of
https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git
synced 2026-05-23 09:02:15 +08:00
Update nodes.py
This commit is contained in:
parent
01a9f8ca51
commit
49e42c5dc6
2
nodes.py
2
nodes.py
@ -148,7 +148,7 @@ class CogVideoPABConfig:
|
|||||||
temporal_broadcast, temporal_threshold_start, temporal_threshold_end, temporal_range,
|
temporal_broadcast, temporal_threshold_start, temporal_threshold_end, temporal_range,
|
||||||
cross_broadcast, cross_threshold_start, cross_threshold_end, cross_range, steps):
|
cross_broadcast, cross_threshold_start, cross_threshold_end, cross_range, steps):
|
||||||
|
|
||||||
#os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
|
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
|
||||||
pab_config = CogVideoXPABConfig(
|
pab_config = CogVideoXPABConfig(
|
||||||
steps=steps,
|
steps=steps,
|
||||||
spatial_broadcast=spatial_broadcast,
|
spatial_broadcast=spatial_broadcast,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user