| license: apache-2.0 | |
| tags: | |
| - finetuned | |
| - quantized | |
| - 4-bit | |
| - gptq | |
| - transformers | |
| - safetensors | |
| - llama | |
| - text-generation | |
| - en | |
| - arxiv:2305.18290 | |
| - arxiv:2106.09685 | |
| - license:mit | |
| - autotrain_compatible | |
| - endpoints_compatible | |
| - text-generation-inference | |
| - region:us | |
| model_name: MoMo-72B-lora-1.8.7-DPO-GPTQ | |
| base_model: moreh/MoMo-72B-lora-1.8.7-DPO | |
| inference: false | |
| model_creator: moreh | |
| pipeline_tag: text-generation | |
| quantized_by: MaziyarPanahi | |
| # Description | |
| [MaziyarPanahi/MoMo-72B-lora-1.8.7-DPO-GPTQ](https://huggingface.co/MaziyarPanahi/MoMo-72B-lora-1.8.7-DPO-GPTQ) is a quantized (GPTQ) version of [moreh/MoMo-72B-lora-1.8.7-DPO](https://huggingface.co/moreh/MoMo-72B-lora-1.8.7-DPO) | |
| ## How to use | |
| ### Install the necessary packages | |
| ``` | |
| pip install --upgrade accelerate auto-gptq transformers | |
| ``` | |
| ### Example Python code | |
| ```python | |
| from transformers import AutoTokenizer, pipeline | |
| from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig | |
| import torch | |
| model_id = "MaziyarPanahi/MoMo-72B-lora-1.8.7-DPO-GPTQ" | |
| quantize_config = BaseQuantizeConfig( | |
| bits=4, | |
| group_size=128, | |
| desc_act=False | |
| ) | |
| model = AutoGPTQForCausalLM.from_quantized( | |
| model_id, | |
| use_safetensors=True, | |
| device="cuda:0", | |
| quantize_config=quantize_config) | |
| tokenizer = AutoTokenizer.from_pretrained(model_id) | |
| pipe = pipeline( | |
| "text-generation", | |
| model=model, | |
| tokenizer=tokenizer, | |
| max_new_tokens=512, | |
| temperature=0.7, | |
| top_p=0.95, | |
| repetition_penalty=1.1 | |
| ) | |
| outputs = pipe("What is a large language model?") | |
| print(outputs[0]["generated_text"]) | |
| ``` |