初始化项目,由ModelHub XC社区提供模型

Model: pvlabs/Chytrej2-Mini-It
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-06-12 16:14:21 +08:00
commit 8a3eb8f8b2
7 changed files with 159236 additions and 0 deletions

104
README.md Normal file
View File

@@ -0,0 +1,104 @@
---
language:
- en
license: apache-2.0
pipeline_tag: text-generation
tags:
- llama
- causal-lm
- finetuned
- chytrej
- instruct
- tiny
- chatml
library_name: transformers
datasets:
- HuggingFaceTB/everyday-conversations-llama3.1-2k
base_model: pvlabs/Chytrej2-Mini
---
# Chytrej2-Mini-It
A fine-tuned version of [Chytrej2-Mini](https://huggingface.co/pvlabs/Chytrej2-Mini) (20M params, LLaMA architecture) trained on conversational data. Don't expect great answers.
Built by [PingVortex Labs](https://github.com/PingVortexLabs).
[![Discord](https://img.shields.io/badge/Discord-5865F2?logo=discord&logoColor=white)](https://discord.gg/5SzkjVJBs2)
---
## Model Details
+ **Parameters:** 20M
+ **Context length:** 1024 tokens
+ **Language:** English only
+ **Format:** ChatML
+ **Base model:** [pvlabs/Chytrej2-Mini](https://huggingface.co/pvlabs/Chytrej2-Mini)
+ **Architecture:** LLaMA
+ **License:** Apache 2.0
---
## Training
Fine-tuned on [HuggingFaceTB/everyday-conversations-llama3.1-2k](https://huggingface.co/datasets/HuggingFaceTB/everyday-conversations-llama3.1-2k) dataset.
---
## Usage
```python
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch
model_path = "pvlabs/Chytrej2-Mini-It"
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(model_path, dtype=torch.float16)
model.eval()
prompt = "<|im_start|>user\nHello<|im_end|>\n<|im_start|>assistant\n"
inputs = tokenizer(prompt, return_tensors="pt")
with torch.no_grad():
output = model.generate(
**inputs,
max_new_tokens=200,
do_sample=True,
temperature=0.7,
top_p=0.9,
eos_token_id=tokenizer.convert_tokens_to_ids("<|im_end|>"),
pad_token_id=tokenizer.eos_token_id,
)
generated = tokenizer.decode(output[0][inputs["input_ids"].shape[1]:], skip_special_tokens=False)
print(generated)
```
---
## Prompt Format (ChatML)
The model uses the standard ChatML format:
```
<|im_start|>user
Your message here<|im_end|>
<|im_start|>assistant
```
For multi-turn, chain turns:
```
<|im_start|>user
Hi!<|im_end|>
<|im_start|>assistant
Hello! How can I help you today?<|im_end|>
<|im_start|>user
What's 2+2?<|im_end|>
<|im_start|>assistant
```
---
*Made by [PingVortex](https://pingvortex.com).*