diff --git a/src/projected_grpo/train.py b/src/projected_grpo/train.py index 1fe49b4..fc2d22a 100644 --- a/src/projected_grpo/train.py +++ b/src/projected_grpo/train.py @@ -415,7 +415,6 @@ def main(cfg: Config) -> int: if not v_hack_path.exists(): from .extract_vhack_grad import extract_v_hack from .pairs import PAIRS as VHACK_PAIRS - from safetensors.torch import save_file logger.info(f"v_hack cache miss at {v_hack_path}; extracting (~5min)...") model.eval() # match standalone extract: deterministic backward, no dropout v_hack_cpu_dict, raw_grads, _diag = extract_v_hack(