ConvAI
Collection
my own lm XD (Mistral based)
• 2 items • Updated • 1
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("CreitinGameplays/ConvAI-9b")
model = AutoModelForCausalLM.from_pretrained("CreitinGameplays/ConvAI-9b")
messages = [
{"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
messages,
add_generation_prompt=True,
tokenize=True,
return_dict=True,
return_tensors="pt",
).to(model.device)
outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))ConvAI-9b is a fine-tuned conversational AI model with 9 billion parameters. It is based on the following models:
The model was fine-tuned on a custom dataset of conversations between an AI assistant and a user. The dataset format followed a specific structure:
<|system|> (system prompt, e.g.: You are a helpful AI language model called ChatGPT, your goal is helping users with their questions) </s> <|user|> (user prompt) </s>
ConvAI-9b is intended for use in conversational AI applications, such as:
| Metrics | Value |
|---|---|
| ARC | 57.50 |
| HellaSwag | 80.34 |
| TruthfulQA | 49.54 |
| Winogrande | 76.24 |
More detailed evaluation here
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="CreitinGameplays/ConvAI-9b") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)