File size: 1,939 Bytes
4bd5172 63fb2a3 |
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 |
---
license: mit
language:
- ru
- en
- uk
- zh
---
# PyroNet-v2
**PyroNet-v2** is a fine-tuned conversational AI model based on [Qwen2.5-3B-Instruct](https://huggingface.co/Qwen/Qwen2.5-3B-Instruct).
It is the successor to **PyroNet-v1.5**, which was built on top of [phi-2](https://huggingface.co/microsoft/phi-2).
Created by **IceL1ghtning (Artyom, Ukraine)**.
---
## π§ Model Details
- **Base model:** Qwen2.5-3B-Instruct
- **Parameters:** ~3B
- **Previous version:** PyroNet-v1.5 (phi-2)
- **Input format:** ChatML (`<|im_start|>role ... <|im_end|>`)
- **Multilingual support:** English, Russian, Ukrainian, and more
---
## π Quick Start
### Installation
```bash
pip install transformers accelerate
```
### Usage Example
```python
from transformers import AutoTokenizer, AutoModelForCausalLM
model_name = "Kenan023214/PyroNet-v2"
# Load tokenizer and model
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
model_name,
device_map="auto"
)
# Example conversation
messages = [
{"role": "user", "content": "Hi! Can you solve the equation x^2 - 5x + 6 = 0?"}
]
# Apply chat template
inputs = tokenizer.apply_chat_template(
messages,
tokenize=True,
add_generation_prompt=True,
return_tensors="pt"
).to(model.device)
# Generate output
outputs = model.generate(
inputs,
max_new_tokens=256,
temperature=0.7,
do_sample=True
)
# Decode and print
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(response)
```
## π Version History
* **PyroNet-v1.5** β based on Microsoft phi-2
* **PyroNet-v2** β upgraded to Qwen2.5-3B-Instruct with improved accuracy and longer context handling
## β οΈ License & Limitations
This model is provided as is.
It must **not** be used for:
harmful or malicious activities
generating unsafe or illegal content
β¦ Created by IceL1ghtning (Artyom, Ukraine) |