Views
No views yet

OpenGCM/Hydrion-SFT) is the instruction-tuned, chat-ready version. The base pretrained checkpoint (no chat formatting) is available at OpenGCM/Hydrion-Base.EleutherAI/gpt-neox-20b BPE tokenizer, extended with <|im_start|> / <|im_end|> special tokens for ChatML formattingdatabricks/databricks-dolly-15k, formatted as ChatML conversations with loss masked to the assistant's response tokens only.lm-evaluation-harness on the base (pre-SFT) checkpoint:| Benchmark | Metric | Score |
|---|---|---|
| BLiMP | acc | 80.08% |
| ARC-Easy | acc | 47.26% |
| ARC-Easy | acc_norm | 43.39% |
| WikiText-2 | byte_perplexity | 2.04 |
| WikiText-2 | bits_per_byte | 1.03 |
| WikiText-2 | word_perplexity | 45.02 |
1import torch
2from transformers import AutoTokenizer, LlamaForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("OPENGCM/Hydrion-SFT")
5model = LlamaForCausalLM.from_pretrained("OPENGCM/Hydrion-SFT", torch_dtype=torch.bfloat16).cuda()
6model.eval()
7
8prompt = "<|im_start|>user\nWhat is the capital of France?<|im_end|>\n<|im_start|>assistant\n"
9inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
10
11with torch.no_grad():
12 output = model.generate(
13 **inputs,
14 max_new_tokens=150,
15 do_sample=True,
16 temperature=0.7,
17 top_p=0.9,
18 repetition_penalty=1.3,
19 no_repeat_ngram_size=3,
20 eos_token_id=tokenizer.convert_tokens_to_ids("<|im_end|>"),
21 )
22
23response = tokenizer.decode(output[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True)
24print(response)