File size: 4,849 Bytes
85723f7 9dc7a0a 85723f7 9dc7a0a 85723f7 9dc7a0a 85723f7 9dc7a0a cf83a12 9dc7a0a cf83a12 85723f7 9dc7a0a 88e1b35 9dc7a0a 88e1b35 bb90af0 9dc7a0a bb90af0 0a5b994 9dc7a0a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 |
---
language:
- en
license: apache-2.0
datasets:
- HuggingFaceH4/ultrachat_200k
- Felladrin/ChatML-ultrachat_200k
base_model: Locutusque/TinyMistral-248M
pipeline_tag: text-generation
widget:
- messages:
- role: system
content:
You are a career counselor. The user will provide you with an individual
looking for guidance in their professional life, and your task is to assist
them in determining what careers they are most suited for based on their skills,
interests, and experience. You should also conduct research into the various
options available, explain the job market trends in different industries, and
advice on which qualifications would be beneficial for pursuing particular fields.
- role: user
content: Heya!
- role: assistant
content: Hi! How may I help you?
- role: user
content:
I am interested in developing a career in software engineering. What
would you recommend me to do?
- messages:
- role: user
content: Morning!
- role: assistant
content: Good morning! How can I help you today?
- role: user
content: Could you give me some tips for becoming a healthier person?
- messages:
- role: user
content: Write the specs of a game about mages in a fantasy world.
- messages:
- role: user
content: Tell me about the pros and cons of social media.
- messages:
- role: system
content:
You are a highly knowledgeable and friendly assistant. Your goal is to
understand and respond to user inquiries with clarity. Your interactions are
always respectful, helpful, and focused on delivering the most accurate information
to the user.
- role: user
content: Hey! Got a question for you!
- role: assistant
content: Sure! What's it?
- role: user
content: What are some potential applications for quantum computing?
inference:
parameters:
max_new_tokens: 250
penalty_alpha: 0.5
top_k: 4
repetition_penalty: 1.12
---
# Locutusque's TinyMistral-248M trained on UltraChat dataset
- Base model: [Locutusque/TinyMistral-248M](https://huggingface.co/Locutusque/TinyMistral-248M) with two additional special tokens (`<|im_start|>` and `<|im_end|>`)
- Dataset: [[ChatML](https://huggingface.co/datasets/Felladrin/ChatML-ultrachat_200k)] [HuggingFaceH4/ultrachat_200k](https://huggingface.co/datasets/HuggingFaceH4/ultrachat_200k)
- License: [Apache License 2.0](https://huggingface.co/Felladrin/TinyMistral-248M-Chat-v1/resolve/main/license.txt)
## Recommended Prompt Format
```
<|im_start|>system
{system_message}<|im_end|>
<|im_start|>user
{user_message}<|im_end|>
<|im_start|>assistant
```
## Recommended Inference Parameters
```yml
penalty_alpha: 0.5
top_k: 4
repetition_penalty: 1.12
```
## Usage Example
```python
from transformers import pipeline
generate = pipeline("text-generation", "Felladrin/TinyMistral-248M-Chat-v1")
messages = [
{
"role": "system",
"content": "You are a highly knowledgeable and friendly assistant. Your goal is to understand and respond to user inquiries with clarity. Your interactions are always respectful, helpful, and focused on delivering the most accurate information to the user.",
},
{
"role": "user",
"content": "Hey! Got a question for you!",
},
{
"role": "assistant",
"content": "Sure! What's it?",
},
{
"role": "user",
"content": "What are some potential applications for quantum computing?",
},
]
prompt = generate.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
output = generate(
prompt,
max_new_tokens=256,
penalty_alpha=0.5,
top_k=4,
repetition_penalty=1.12,
)
print(output[0]["generated_text"])
```
## How it was trained
This model was trained with [SFTTrainer](https://huggingface.co/docs/trl/main/en/sft_trainer) using the following settings:
| Hyperparameter | Value |
| :--------------------- | :-------------------------------------------- |
| Learning rate | 2e-5 |
| Total train batch size | 16 |
| Max. sequence length | 2048 |
| Weight decay | 0 |
| Warmup ratio | 0.1 |
| Optimizer | Adam with betas=(0.9,0.999) and epsilon=1e-08 |
| Scheduler | cosine |
| Seed | 42 |
|