112 lines
3.5 KiB
Markdown
112 lines
3.5 KiB
Markdown
---
|
|
base_model: MaziyarPanahi/Llama-3-8B-Instruct-DPO-v0.1
|
|
library_name: transformers
|
|
tags:
|
|
- 4-bit
|
|
- AWQ
|
|
- text-generation
|
|
- autotrain_compatible
|
|
- endpoints_compatible
|
|
- axolotl
|
|
- finetune
|
|
- facebook
|
|
- meta
|
|
- pytorch
|
|
- llama
|
|
- llama-3
|
|
language:
|
|
- en
|
|
pipeline_tag: text-generation
|
|
license: other
|
|
license_name: llama3
|
|
license_link: LICENSE
|
|
inference: false
|
|
model_creator: MaziyarPanahi
|
|
model_name: Llama-3-8B-Instruct-DPO-v0.1
|
|
quantized_by: Suparious
|
|
datasets:
|
|
- mlabonne/chatml-OpenHermes2.5-dpo-binarized-alpha
|
|
---
|
|
# MaziyarPanahi/Llama-3-8B-Instruct-DPO-v0.1 AWQ
|
|
|
|
- Model creator: [MaziyarPanahi](https://huggingface.co/MaziyarPanahi)
|
|
- Original model: [Llama-3-8B-Instruct-DPO-v0.1](https://huggingface.co/MaziyarPanahi/Llama-3-8B-Instruct-DPO-v0.1)
|
|
|
|
<img src="./llama-3-merges.webp" alt="Llama-3 DPO Logo" width="500" style="margin-left:'auto' margin-right:'auto' display:'block'"/>
|
|
|
|
This model is a fine-tune (DPO) of `meta-llama/Meta-Llama-3-8B-Instruct` model.
|
|
|
|
## How to use
|
|
|
|
This model uses `ChatML` prompt template:
|
|
|
|
```
|
|
<|im_start|>system
|
|
{System}
|
|
<|im_end|>
|
|
<|im_start|>user
|
|
{User}
|
|
<|im_end|>
|
|
<|im_start|>assistant
|
|
{Assistant}
|
|
````
|
|
|
|
### Install the necessary packages
|
|
|
|
```bash
|
|
pip install --upgrade autoawq autoawq-kernels
|
|
```
|
|
|
|
### Example Python code
|
|
|
|
```python
|
|
from awq import AutoAWQForCausalLM
|
|
from transformers import AutoTokenizer, TextStreamer
|
|
|
|
model_path = "solidrust/Llama-3-8B-Instruct-DPO-v0.1-AWQ"
|
|
system_message = "You are Llama-3-8B-Instruct-DPO-v0.1, incarnated as a powerful AI. You were created by MaziyarPanahi."
|
|
|
|
# Load model
|
|
model = AutoAWQForCausalLM.from_quantized(model_path,
|
|
fuse_layers=True)
|
|
tokenizer = AutoTokenizer.from_pretrained(model_path,
|
|
trust_remote_code=True)
|
|
streamer = TextStreamer(tokenizer,
|
|
skip_prompt=True,
|
|
skip_special_tokens=True)
|
|
|
|
# Convert prompt to tokens
|
|
prompt_template = """\
|
|
<|im_start|>system
|
|
{system_message}<|im_end|>
|
|
<|im_start|>user
|
|
{prompt}<|im_end|>
|
|
<|im_start|>assistant"""
|
|
|
|
prompt = "You're standing on the surface of the Earth. "\
|
|
"You walk one mile south, one mile west and one mile north. "\
|
|
"You end up exactly where you started. Where are you?"
|
|
|
|
tokens = tokenizer(prompt_template.format(system_message=system_message,prompt=prompt),
|
|
return_tensors='pt').input_ids.cuda()
|
|
|
|
# Generate output
|
|
generation_output = model.generate(tokens,
|
|
streamer=streamer,
|
|
max_new_tokens=512)
|
|
```
|
|
|
|
### About AWQ
|
|
|
|
AWQ is an efficient, accurate and blazing-fast low-bit weight quantization method, currently supporting 4-bit quantization. Compared to GPTQ, it offers faster Transformers-based inference with equivalent or better quality compared to the most commonly used GPTQ settings.
|
|
|
|
AWQ models are currently supported on Linux and Windows, with NVidia GPUs only. macOS users: please use GGUF models instead.
|
|
|
|
It is supported by:
|
|
|
|
- [Text Generation Webui](https://github.com/oobabooga/text-generation-webui) - using Loader: AutoAWQ
|
|
- [vLLM](https://github.com/vllm-project/vllm) - version 0.2.2 or later for support for all model types.
|
|
- [Hugging Face Text Generation Inference (TGI)](https://github.com/huggingface/text-generation-inference)
|
|
- [Transformers](https://huggingface.co/docs/transformers) version 4.35.0 and later, from any code or client that supports Transformers
|
|
- [AutoAWQ](https://github.com/casper-hansen/AutoAWQ) - for use from Python code
|