Views
No views yet

OpenGCM/Hydrion-Base) is the base model, non-chat ready version. The instruction model (chat formatting) is available at OpenGCM/Hydrion-SFT.EleutherAI/gpt-neox-20blm-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-Base")
5model = LlamaForCausalLM.from_pretrained("OPENGCM/Hydrion-Base", torch_dtype=torch.bfloat16).cuda()
6model.eval()
7
8prompt = "What is the capital of"
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)