--- license: apache-2.0 tags: - finetuned - quantized - 4-bit - gptq - transformers - safetensors - mistral - text-generation - openchat - C-RLFT - arxiv:2309.11235 - arxiv:2303.08774 - base_model:mistralai/Mistral-7B-v0.1 - license:apache-2.0 - autotrain_compatible - endpoints_compatible - has_space - text-generation-inference - region:us model_name: openchat-3.5-0106-GPTQ base_model: openchat/openchat-3.5-0106 inference: false model_creator: openchat pipeline_tag: text-generation quantized_by: MaziyarPanahi --- # Description [MaziyarPanahi/openchat-3.5-0106-GPTQ](https://huggingface.co/MaziyarPanahi/openchat-3.5-0106-GPTQ) is a quantized (GPTQ) version of [openchat/openchat-3.5-0106](https://huggingface.co/openchat/openchat-3.5-0106) ## 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/openchat-3.5-0106-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"]) ```