diff --git a/tests/parity/hf_snapshot.h b/tests/parity/hf_snapshot.h index 7d4018616..82f076b8b 100644 --- a/tests/parity/hf_snapshot.h +++ b/tests/parity/hf_snapshot.h @@ -41,19 +41,6 @@ inline constexpr const char* kQwen27NvfP4Revision = inline constexpr const char* kNemotron35LightningNvfP4Revision = "29f2d1746d8f41e316523194b19018707749b1b1"; -// The Nemotron-3.5-Lightning gate model (#517). Unlike the Qwen pins above, -// this one is NOT in the HF cache: it is staged on the NAS as a `local_dir` -// snapshot at `$CHECKPOINT_ROOT/nemotron-3.5-lightning-30b-nvfp4`, so there is -// no `models--org--name/snapshots/` layout to resolve. The env override is -// therefore the ONLY reachable path, and the cache spelling below exists so the -// revision still names what the golden belongs to. Absent env var => "" => the -// caller emits its loud SKIP, which is the intended behavior off the gate host. -inline std::string Nemotron35LightningSnapshot() { - return HfSnapshot("models--nvidia--NVIDIA-Nemotron-3.5-Lightning-30B-A3B-NVFP4", - kNemotron35LightningNvfP4Revision, - "VT_NEMOTRON35_SNAPSHOT"); -} - // Snapshot directory for `` at `revision`, or "" when it is not cached // (the caller then emits its loud SKIP). `env_override`, when set and non-empty, // names an explicit snapshot directory for a deliberate different-checkpoint @@ -78,6 +65,19 @@ inline std::string HfSnapshot(const char* repo_dir, const char* revision, return snap.string(); } +// The Nemotron-3.5-Lightning gate model (#517). Unlike the Qwen pins above, +// this one is NOT in the HF cache: it is staged on the NAS as a `local_dir` +// snapshot at `$CHECKPOINT_ROOT/nemotron-3.5-lightning-30b-nvfp4`, so there is +// no `models--org--name/snapshots/` layout to resolve. The env override is +// therefore the ONLY reachable path, and the cache spelling below exists so the +// revision still names what the golden belongs to. Absent env var => "" => the +// caller emits its loud SKIP, which is the intended behavior off the gate host. +inline std::string Nemotron35LightningSnapshot() { + return HfSnapshot("models--nvidia--NVIDIA-Nemotron-3.5-Lightning-30B-A3B-NVFP4", + kNemotron35LightningNvfP4Revision, + "VT_NEMOTRON35_SNAPSHOT"); +} + // The 27B NVFP4 gate model, pinned to the goldens' revision. inline std::string Qwen27NvfP4Snapshot() { return HfSnapshot("models--unsloth--Qwen3.6-27B-NVFP4",