from awq import AutoAWQForCausalLM
from transformers import AutoTokenizer, TextStreamer
model_path = "solidrust/bagel-7b-v0.4-AWQ"
system_message = "You are Bagel, incarnated a powerful AI with everything."# 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.
bagel-7b-v0.4-AWQ huggingface.co is an AI model on huggingface.co that provides bagel-7b-v0.4-AWQ's model effect (), which can be used instantly with this solidrust bagel-7b-v0.4-AWQ model. huggingface.co supports a free trial of the bagel-7b-v0.4-AWQ model, and also provides paid use of the bagel-7b-v0.4-AWQ. Support call bagel-7b-v0.4-AWQ model through api, including Node.js, Python, http.
bagel-7b-v0.4-AWQ huggingface.co is an online trial and call api platform, which integrates bagel-7b-v0.4-AWQ's modeling effects, including api services, and provides a free online trial of bagel-7b-v0.4-AWQ, you can try bagel-7b-v0.4-AWQ online for free by clicking the link below.
solidrust bagel-7b-v0.4-AWQ online free url in huggingface.co:
bagel-7b-v0.4-AWQ is an open source model from GitHub that offers a free installation service, and any user can find bagel-7b-v0.4-AWQ on GitHub to install. At the same time, huggingface.co provides the effect of bagel-7b-v0.4-AWQ install, users can directly use bagel-7b-v0.4-AWQ installed effect in huggingface.co for debugging and trial. It also supports api for free installation.