Address WAN Spectrum review feedback
This commit is contained in:
@@ -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
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user