amirMohammadi/Dorna-Llama3-8B-Instruct-Quantized4Bit

Model Details

This Repository is a 4-bit quantized version of Dorna-Llama3-8B-Instruct model for efficient memory usage. Dorna model is a decoder-only model, specifically trained/fine-tuned on Persian data. Flash Attention 2 is also integrated for faster inference.

Benefits

• —Reduced Memory Usage: 4-bit quantization lowers memory requirements.
• —Faster Inference: Flash Attention 2 speeds up processing.
• —Easy Deployment: No need for additional libraries like LlamaCPP or Candle.
• —Ready to Use: Compatible with Langchain, Haystack, LlamaIndex 2, and more.
• —Google Colab Friendly: Can run on Google Colab free tier with T4 GPU (less than 15 GB of GPU RAM).

How to use

You can run conversational inference using the Transformers Auto classes with the generate() function. Let's look at an example.

import torch
import transformers
from transformers import AutoTokenizer, AutoModelForCausalLM

model_path = "amirMohammadi/Dorna-Llama3-8B-Instruct-Quantized4Bit"

tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(
    model_path,
    torch_dtype=torch.bfloat16,
    device_map="auto",
)

messages = [
    {"role": "system",
     "content": "You are a helpful Persian assistant. Please answer questions in the asked language."},
    {"role": "user", "content": "اصفهان بزرگ تر است یا قم؟"},
]

input_ids = tokenizer.apply_chat_template(
    messages,
    add_generation_prompt=True,
    return_tensors="pt"
).to(model.device)

terminators = [
    tokenizer.eos_token_id,
    tokenizer.convert_tokens_to_ids("<|eot_id|>")
]

outputs = model.generate(
    input_ids,
    max_new_tokens=256,
    eos_token_id=terminators,
    do_sample=True,
    temperature=0.6,
    top_p=0.9,
)
response = outputs[0][input_ids.shape[-1]:]
print(tokenizer.decode(response, skip_special_tokens=True))

Evaluation of Non-Quantized version

This model is evaluated on questions across various tasks, including Boolean Questions, Code Generation, Long Response, Math, News QA, Paraphrasing, General Knowledge, and Summarization. Most categories typically have two main difficulty levels: Hard and Easy.

Both human evaluation and automatic evaluation (with GPT-4 as the judge) are performed.

In both tables, Dorna-8B-it is used as an abbreviated form of Dorna-Llama3-8B-Instruct.

Overall human evaluation results are as follows:

Category-based human evaluation results are as follows:

Win/Lose/Tie % is reported for each category.

-->

<div style="overflow-x: auto;"> <table> <thead> <tr style="vertical-align: middle;"> <th style="white-space: nowrap; vertical-align: middle;"><strong>Model Pairs</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>Parameters</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>Bool Complex</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>Bool Easy</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>Code Gen</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>General Long Response</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>Historical Long Response</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>Math Complex</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>Math Easy</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>News QA Complex</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>News QA Easy</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>Paraphrasing</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>General Knowledge Easy</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>General Knowledge Hard</strong></th> <th style="white-space: nowrap; vertical-align: middle;"><strong>Summarization</strong></th> </tr> </thead> <tbody> <tr> <td style="white-space: nowrap; vertical-align: middle;">Dorna-8B-it <strong>vs.</strong> Meta-Llama-3-8B-Instruct</td> <td>8B</td> <td>0.25/0.25/0.5</td> <td>0.28/<strong>0.35</strong>/0.38</td> <td><strong>0.6</strong>/0.1/0.3</td> <td><strong>0.8</strong>/0.08/0.12</td> <td><strong>0.4</strong>/0.3/0.3</td> <td><strong>0.28</strong>/0.08/0.65</td> <td><strong>0.47</strong>/0.00/0.53</td> <td><strong>0.55</strong>/0.07/0.38</td> <td><strong>0.43</strong>/0.15/0.42</td> <td><strong>0.1</strong>/0.05/0.85</td> <td><strong>0.31</strong>/0.2/0.49</td> <td><strong>0.59</strong>/0.13/0.28</td> <td><strong>0.28</strong>/0.2/0.53</td> </tr> <tr> <td style="white-space: nowrap; vertical-align: middle;">Dorna-8B-it <strong>vs.</strong> GPT 3.5 turbo-1106</td> <td>N.A.</td> <td>0.35/0.35/0.3</td> <td>0.3/0.3/0.4</td> <td>0.1/<strong>0.3</strong>/.06</td> <td>0.2/<strong>0.45</strong>/0.35</td> <td><strong>0.46</strong>/0.27/0.27</td> <td><strong>0.25</strong>/0.1/0.65</td> <td>0.05/<strong>0.1</strong>/0.85</td> <td>0.12/<strong>0.35</strong>/0.53</td> <td><strong>0.15</strong>/0.1/0.75</td> <td><strong>0.25</strong>/0.15/0.6</td> <td>0.3/<strong>0.32</strong>/0.38</td> <td>0.22/<strong>0.53</strong>/0.25</td> <td>0.35/<strong>0.55</strong>/0.1</td> </tr> <tr> <td style="white-space: nowrap; vertical-align: middle;">Dorna-8B-it <strong>vs.</strong> Persian Mind</td> <td>7B</td> <td><strong>0.47</strong>/0.25/0.28</td> <td><strong>0.57</strong>/0.15/0.28</td> <td><strong>0.9</strong>/0.1/0.0</td> <td><strong>0.82</strong>/0.08/0.1</td> <td><strong>0.4</strong>/0.17/0.42</td> <td><strong>0.3</strong>/0.0/0.7</td> <td><strong>0.22</strong>/0.08/0.7</td> <td><strong>0.72</strong>/0.07/0.2</td> <td><strong>0.7</strong>/0.0/0.3</td> <td><strong>0.7</strong>/0.05/0.25</td> <td><strong>0.51</strong>/0.12/0.37</td> <td><strong>0.61</strong>/0.1/0.29</td> <td><strong>0.93</strong>/0.0/0.07</td> </tr> </tbody> </table> </div>

Automatic evaluation results are as follows:

Contact us

If you have any questions regarding this model, you can reach us via the community on Hugging Face.