metadata

license: cc-by-nc-4.0

🌞🚀 SOLAR-math-10.7x2_19B

Merge of two Solar-10.7B instruct finetunes.

$solar$

Runs on 13GB of VRAM in 4bit

🌅 Code Example

Example also available in colab

from transformers import AutoModelForCausalLM, AutoTokenizer

def generate_response(prompt):
    """
    Generate a response from the model based on the input prompt.

    Args:
    prompt (str): Prompt for the model.

    Returns:
    str: The generated response from the model.
    """
    # Tokenize the input prompt
    inputs = tokenizer(prompt, return_tensors="pt")
    
    # Generate output tokens
    outputs = model.generate(**inputs, max_new_tokens=512, eos_token_id=tokenizer.eos_token_id, pad_token_id=tokenizer.pad_token_id)

    # Decode the generated tokens to a string
    response = tokenizer.decode(outputs[0], skip_special_tokens=True)

    return response


# Load the model and tokenizer
model_id = "macadeliccc/SOLAR-math-2x10.7B-v0.2"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, load_in_4bit=True)

prompt = "Explain the proof of Fermat's Last Theorem and its implications in number theory."


print("Response:")
print(generate_response(prompt), "\n")

🏆 Evaluations

Tasks	Version	Filter	Metric	Value		Stderr
arc_challenge	Yaml	none	acc	0.6067	±	0.0143
		none	acc_norm	0.6263	±	0.0141
arc_easy	Yaml	none	acc	0.8211	±	0.0079
		none	acc_norm	0.8001	±	0.0082
boolq	Yaml	none	acc	0.8557	±	0.0061
hellaswag	Yaml	none	acc	0.6695	±	0.0047
		none	acc_norm	0.8484	±	0.0036
openbookqa	Yaml	none	acc	0.3420	±	0.0212
		none	acc_norm	0.4520	±	0.0223
piqa	Yaml	none	acc	0.7949	±	0.0094
		none	acc_norm	0.8058	±	0.0092
winogrande	Yaml	none	acc	0.7372	±	0.0124

📚 Citations

@misc{kim2023solar,
      title={SOLAR 10.7B: Scaling Large Language Models with Simple yet Effective Depth Up-Scaling}, 
      author={Dahyun Kim and Chanjun Park and Sanghoon Kim and Wonsung Lee and Wonho Song and Yunsu Kim and Hyeonwoo Kim and Yungi Kim and Hyeonju Lee and Jihoo Kim and Changbae Ahn and Seonghoon Yang and Sukyung Lee and Hyunbyung Park and Gyoungjin Gim and Mikyoung Cha and Hwalsuk Lee and Sunghun Kim},
      year={2023},
      eprint={2312.15166},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}