File size: 1,720 Bytes
17c042b
 
 
810fd7e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c3b75d3
810fd7e
6998275
 
810fd7e
 
 
 
 
 
 
 
 
 
 
 
 
6998275
 
 
 
 
 
 
 
 
 
 
 
 
d74c89d
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
---
license: mit
---
<div align="center">

# πŸ€– SmallDront-60M

*A lightweight conversational AI model designed for natural small talk*

[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
[![Hugging Face](https://img.shields.io/badge/πŸ€—%20Hugging%20Face-Model-blue)](https://huggingface.co/)
[![Parameters](https://img.shields.io/badge/Parameters-60M-green)]()
[![Format](https://img.shields.io/badge/Format-F32-orange)]()

</div>

---

## πŸ“ Overview

**SmallDront-60M** is a compact language model specifically fine-tuned for engaging small talk conversations. With 60 million parameters in F32 precision, it strikes a balance between performance and efficiency β€” delivering coherent, natural dialogue without the overhead of larger models.

> *Compared to its predecessor (SmallDront-20M), this model properly ends its responses and hallucinates significantly less.*

---

## ✨ Features

| Feature | Detail |
|---------|--------|
| 🧠 **Architecture** | Transformer-based |
| πŸ”’ **Parameters** | +-60,000,000 |
| πŸ“ **Precision** | F32 (Full float32) |
| πŸ”€ **Tokenizer** | **GPT-2 Tokenizer** |
| πŸ“¦ **Format** | Hugging Face  |
| 🏷️ **Special Tokens** | `<\|user\|>` `<\|assistant\|>` |

---

## πŸŽ“ Training

The model was trained on the **[VDC-team/DialoguesEN-4k](https://huggingface.co/datasets/VDC-team/DialoguesEN-4k)** dataset until reaching a loss of **0.9**.

---

## πŸ’¬ Example Conversations

*Temperature: 0.3*

```text
You: Who are you?
Assistant: Just a friendly chat. You?

You: Hello!
Assistant: Hey! What's a place you feel at?

You: Where you?
Assistant: Hi, for a chat. You?

```

**use.py** - use model example