Fix UnboundLocalError: eff_key not yet assigned when VF loads

Used refinement_key_input/refinement_key directly instead of eff_key
which is assigned ~340 lines later in refine_v2.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
digital-garbage
2026-05-30 13:49:32 +03:00
co-authored by Claude Sonnet 4.6
parent 28ef855893
commit d8763da4a7
+3 -2
View File
@@ -11539,13 +11539,14 @@ class FunPackVideoRefinerV2(FunPackVideoRefiner):
# Load value function for conditioning memory modulation (confidence + gradient boost)
_vf_for_memory = None
if not learning_mode and not prompt_only_mode and eff_key:
_eff_key_early = str(refinement_key_input or refinement_key or "").strip()
if not learning_mode and not prompt_only_mode and _eff_key_early:
try:
try:
from .value_function import OnlineValueFunction as _OVF
except ImportError:
from value_function import OnlineValueFunction as _OVF
_vf_path = refinement_state_path(eff_key, "value_fn", prefix="refine_v2", extension="pt")
_vf_path = refinement_state_path(_eff_key_early, "value_fn", prefix="refine_v2", extension="pt")
if os.path.exists(_vf_path):
import torch as _torch
with _torch.inference_mode(False):