Skip to content

Commit 6490af1

Browse files
committed
fixed standin with boost off
1 parent 336e941 commit 6490af1

2 files changed

Lines changed: 7 additions & 7 deletions

File tree

models/wan/modules/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1306,7 +1306,7 @@ def forward(
13061306
if standin_ref is not None:
13071307
standin_cache_enabled = False
13081308
kwargs["standin_phase"] = 2
1309-
if (current_step == 0 or not standin_cache_enabled) and x_id == 0:
1309+
if current_step == 0 or not standin_cache_enabled :
13101310
standin_x = self.patch_embedding(standin_ref).to(modulation_dtype).flatten(2).transpose(1, 2)
13111311
standin_e = self.time_embedding( sinusoidal_embedding_1d(self.freq_dim, torch.zeros_like(t)).to(modulation_dtype) )
13121312
standin_e0 = self.time_projection(standin_e).unflatten(1, (6, self.dim)).to(e.dtype)
@@ -1453,7 +1453,7 @@ def forward(
14531453
return [None] * len(x_list)
14541454

14551455
if standin_x is not None:
1456-
if not standin_cache_enabled and x_id ==0 : get_cache("standin").clear()
1456+
if not standin_cache_enabled: get_cache("standin").clear()
14571457
standin_x = block(standin_x, context = None, grid_sizes = None, e= standin_e0, freqs = standin_freqs, standin_phase = 1)
14581458

14591459
if slg_layers is not None and block_idx in slg_layers:

wgp.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,8 @@
11
import os
2+
# # os.environ.pop("TORCH_LOGS", None) # make sure no env var is suppressing/overriding
3+
# os.environ["TORCH_LOGS"]= "recompiles"
4+
import torch._logging as tlog
5+
# tlog.set_logs(recompiles=True, guards=True, graph_breaks=True)
26
import time
37
import sys
48
import threading
@@ -55,7 +59,7 @@
5559
PROMPT_VARS_MAX = 10
5660

5761
target_mmgp_version = "3.5.10"
58-
WanGP_version = "8.2"
62+
WanGP_version = "8.21"
5963
settings_version = 2.27
6064
max_source_video_frames = 3000
6165
prompt_enhancer_image_caption_model, prompt_enhancer_image_caption_processor, prompt_enhancer_llm_model, prompt_enhancer_llm_tokenizer = None, None, None, None
@@ -4308,10 +4312,6 @@ def generate_video(
43084312
model_filename,
43094313
mode,
43104314
):
4311-
# import os
4312-
# os.environ.pop("TORCH_LOGS", None) # make sure no env var is suppressing/overriding
4313-
# import torch._logging as tlog
4314-
# tlog.set_logs(recompiles=True, guards=True, graph_breaks=True)
43154315

43164316

43174317

0 commit comments

Comments
 (0)