chore: 修改文件夹结构
This commit is contained in:
parent
b1527d9575
commit
4ead0a20cf
20
README.md
20
README.md
|
|
@ -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 pre‑processed 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
|
||||
|
||||
# Auto‑regressive generation
|
||||
python demo/generate_ar.py
|
||||
python scripts/demo/generate_ar.py
|
||||
```
|
||||
|
||||
Watch a video walkthrough on [bilibili](https://www.bilibili.com/video/BV1z5RPYHEkd).
|
||||
|
|
|
|||
|
|
@ -75,7 +75,7 @@ pip install -e ".[dev]"
|
|||
#### 训练模型
|
||||
|
||||
```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
|
||||
|
|
@ -84,25 +84,25 @@ python tools/train.py \
|
|||
#### 文本生成
|
||||
|
||||
```bash
|
||||
python tools/generate.py --param_path=/path/to/param_path
|
||||
python scripts/tools/generate.py --param_path=/path/to/param_path
|
||||
```
|
||||
|
||||
#### 演示
|
||||
|
||||
查看 `demo/` 文件夹中的演示:
|
||||
查看 `scripts/demo/` 文件夹中的演示:
|
||||
|
||||
```bash
|
||||
# 下载预处理数据(运行演示前必需)
|
||||
python demo/download.py
|
||||
python scripts/demo/download.py
|
||||
|
||||
# 交互式流式聊天
|
||||
python demo/stream_chat.py
|
||||
python scripts/demo/stream_chat.py
|
||||
|
||||
# 批量生成
|
||||
python demo/generate_batch.py
|
||||
python scripts/demo/generate_batch.py
|
||||
|
||||
# 自回归生成
|
||||
python demo/generate_ar.py
|
||||
python scripts/demo/generate_ar.py
|
||||
```
|
||||
|
||||
观看 [bilibili](https://www.bilibili.com/video/BV1z5RPYHEkd) 上的视频演示。
|
||||
|
|
|
|||
Loading…
Reference in New Issue