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

Model: JinyiHan/JET-1.5B
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-10-08 02:17:17 +08:00
commit caca607543
15 changed files with 368 additions and 0 deletions

38
README.md Normal file
View File

@@ -0,0 +1,38 @@
---
license: apache-2.0
tags:
- text-generation
- causal-lm
- reasoning
library_name: transformers
---
# JET-1.5B
JET-1.5B is designed to improve the efficient reasoning of LLMs by training the base **DeepSeek-Distill-Qwen-1.5B** model with a reinforcement learning framework. Through this training, the model learns to generate high-quality reasoning steps while minimizing unnecessary computation and token usage.
# Training Code
Our training pipeline is available on GitHub: [Just-Enough-Think](https://github.com/JinyiHan99/Just-Enough-Think/)
The repository contains scripts for:
- RL-based fine-tuning
- Evaluation and benchmarking
## Chat Template
```python
def build_JET_chat_template(question, tokenizer):
system_prompt = (
"You are a helpful AI assistant. A conversation takes place between the User "
"and the Assistant. The User asks a question, and the Assistant solves it.\n"
"Please help me solve this question. Wrap only the final answer in \\boxed{}."
)
return tokenizer.apply_chat_template(
[
{"role": "system", "content": system_prompt},
{"role": "user", "content": question}
],
tokenize=False,
add_generation_prompt=True
)