How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="rxavier/Taurus-7B-1.0")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("rxavier/Taurus-7B-1.0")
model = AutoModelForCausalLM.from_pretrained("rxavier/Taurus-7B-1.0")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
Quick Links

Taurus 7B 1.0

image/png

Description

Taurus is an OpenHermes 2.5 finetune using the Economicus dataset, an instruct dataset synthetically generated from Economics PhD textbooks.

The model was trained for 2 epochs (QLoRA) using axolotl. The exact config I used can be found here.

Open LLM Leaderboard Evaluation Results

Detailed results can be found here

Metric Value
Avg. 66.40
AI2 Reasoning Challenge (25-Shot) 63.57
HellaSwag (10-Shot) 83.64
MMLU (5-Shot) 63.50
TruthfulQA (0-shot) 50.21
Winogrande (5-shot) 78.14
GSM8k (5-shot) 59.36

Prompt format

Taurus uses ChatML.

<|im_start|>system
System message
<|im_start|>user
User message<|im_end|>
<|im_start|>assistant

Usage

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig


model_id = "rxavier/Taurus-7B-1.0"
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    torch_dtype=torch.bfloat16, #torch.float16 for older GPUs
    device_map="auto", # Requires having accelerate installed, useful in places like Colab with limited VRAM
)
tokenizer = AutoTokenizer.from_pretrained(model_id)
generation_config = GenerationConfig(
                bos_token_id=tokenizer.bos_token_id,
                eos_token_id=tokenizer.eos_token_id,
                pad_token_id=tokenizer.pad_token_id,
            )

system_message = "You are an expert in economics with PhD level knowledge. You are helpful, give thorough and clear explanations, and use equations and formulas where needed."
prompt = "Give me latex formulas for extended euler equations"

messages = [{"role": "system",
             "content": system_message},
            {"role": "user",
             "content": prompt}]
tokens = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to("cuda")

with torch.no_grad():
    outputs = model.generate(inputs=tokens, generation_config=generation_config, max_length=512)
print(tokenizer.decode(outputs.cpu().tolist()[0]))

GGUF quants

You can find GGUF quants for llama.cpp here.

Downloads last month
54
Safetensors
Model size
7B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for rxavier/Taurus-7B-1.0

Finetuned
(85)
this model
Quantizations
3 models

Evaluation results