Address WAN Spectrum review feedback

This commit is contained in:
xmarre
2026-05-13 03:38:46 +02:00
parent f28c87e0d5
commit 6a081b4e7b
2 changed files with 6 additions and 4 deletions
+4 -1
View File
@@ -247,7 +247,10 @@ class TIDEModelWrapper:
if has_tide_wan_options(transformer_options):
inject_tide_wan_options(transformer_options, self.config, timestep=args.get("timestep"), source="model_function_wrapper")
except Exception as exc:
if self.config.debug:
if not getattr(self, "_wan_injection_warned", False):
_LOG.warning("TIDE WAN transformer_options injection failed and was skipped: %s", exc, exc_info=True)
self._wan_injection_warned = True
elif self.config.debug:
_LOG.exception("TIDE WAN transformer_options injection failed and was skipped: %s", exc)
c["transformer_options"] = transformer_options
+2 -3
View File
@@ -383,8 +383,7 @@ def _resolve_apply_model_outer(apply_model: Any) -> Any:
return None
def _resolve_apply_model_wan_inner(apply_model: Any) -> Any:
outer = _resolve_apply_model_outer(apply_model)
def _resolve_apply_model_wan_inner(outer: Any) -> Any:
for candidate in _wan_candidates_from_outer(outer):
if _looks_like_wan_inner(candidate):
return candidate
@@ -463,7 +462,7 @@ def prepare_tide_wan_apply_model(
return
outer = _resolve_apply_model_outer(apply_model)
inner = _resolve_apply_model_wan_inner(apply_model)
inner = _resolve_apply_model_wan_inner(outer)
if not _looks_like_wan_inner(inner):
return