metadata
license: apache-2.0
language:
- en
- zh
library_name: transformers
widget:
- text: <s> [|User|] Hi π </s>[|Assistant|]
This is a quantized version of GeneZC/MiniChat-2-3B
Optimum quantization using the command:
optimum-cli inc quantize --model GeneZC/MiniChat-2-3B --output ./mnch
Usage example:
from optimum.intel import INCModelForCausalLM
from transformers import AutoTokenizer, pipeline, AutoModelForCausalLM
import torch
model_id = "Mihaiii/MiniChat-2-3B-optimum-intel"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = INCModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
messages = [
{
"role": "system",
"content": "You are a friendly chatbot who always responds in the style of a pirate",
},
{"role": "user", "content": "How many helicopters can a human eat in one sitting?"},
]
prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.0001, repetition_penalty=1.2)
print(outputs[0]["generated_text"])