chore: 修改文件夹结构

This commit is contained in:
2026-03-31 10:14:08 +08:00
parent b1527d9575
commit 4ead0a20cf
10 changed files with 14 additions and 20 deletions
+7 -13
View File
@@ -2,12 +2,6 @@
<!-- <img src="assets/images/project_logo.png" width="auto" alt="Logo"> -->
<h1>AstrAI</h1>
<div>
<a href="#english">English</a> •
<a href="assets/docs/README-zh-CN.md">中文</a>
</div>
<p>
<strong>A lightweight Transformer training & inference framework</strong>
</p>
@@ -80,7 +74,7 @@ pip install -e ".[dev]"
#### Train a Model
```bash
python tools/train.py \
python scripts/tools/train.py \
--train_type=seq \
--data_root_path=/path/to/dataset \
--param_path=/path/to/param_path
@@ -89,25 +83,25 @@ python tools/train.py \
#### Generate Text
```bash
python tools/generate.py --param_path=/path/to/param_path
python scripts/tools/generate.py --param_path=/path/to/param_path
```
#### Demo
Check out the demos in the `demo/` folder:
Check out the demos in the `scripts/demo/` folder:
```bash
# Download preprocessed data (required before running demos)
python demo/download.py
python scripts/demo/download.py
# Interactive streaming chat
python demo/stream_chat.py
python scripts/demo/stream_chat.py
# Batch generation
python demo/generate_batch.py
python scripts/demo/generate_batch.py
# Autoregressive generation
python demo/generate_ar.py
python scripts/demo/generate_ar.py
```
Watch a video walkthrough on [bilibili](https://www.bilibili.com/video/BV1z5RPYHEkd).