Upload v24_grpo_chain_v2.py with huggingface_hub

v24_grpo_chain_v2.py

@@ -186,7 +186,7 @@ def main():
     if len(rows) < args.group_size:
         raise RuntimeError("Too few valid sequential chain rows")
 
-    cfg = preset_config(args.model_size, vocab_size=tok.vocab_size, block_size=args.block_size, dropout=0.0, pcs_a=0.8309193524, pcs_b=0.0)
+    cfg = preset_config(args.model_size, vocab_size=len(tok), block_size=args.block_size, dropout=0.0, pcs_a=0.8309193524, pcs_b=0.0)
     model = ScugnizzDecoder(cfg).to(device=device, dtype=dtype)
     state = load_checkpoint(args.checkpoint, "cpu")
     if isinstance(state, dict) and "model" in state: