Views
No views yet
1# Load model directly
2from transformers import AutoTokenizer, AutoModelForCausalLM
3tokenizer = AutoTokenizer.from_pretrained("EpistemeAI/Hercules1-8B-E2B-it")
4model = AutoModelForCausalLM.from_pretrained("EpistemeAI/Hercules1-8B-E2B-it")
5messages = [
6 {"role": "user", "content": "Write me a Python function to calculate the nth fibonacci number.<think></think>"},
7]
8inputs = tokenizer.apply_chat_template(
9 messages,
10 add_generation_prompt=True,
11 tokenize=True,
12 return_dict=True,
13 return_tensors="pt",
14).to(model.device)
15outputs = model.generate(**inputs, max_new_tokens=1024)
16print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))| Tasks | Version | Filter | n-shot | Metric | Value | |
|---|---|---|---|---|---|---|
| gpqa_diamond_zeroshot | 1 | none | 0 | acc/acc_norm | ↑ | 0.3097 |