update

Oneflow-Inc · Sep 14, 2024 · 3fa6811 · 3fa6811
2 parents 3d57656 + 8cd8b4b
commit 3fa6811
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/projects/Aquila/README.md b/projects/Aquila/README.md
@@ -8,8 +8,20 @@
 python projects/Aquila/pipeline.py --model_path=/root/models/Aquila-7B --mode=huggingface
 ```
 
-### 训练
--- data preparation
+- npu
+
+```bash
+python projects/Aquila/pipeline.py --model_path=/data0/hf_models/Aquila-7B --mode=huggingface --device=npu
+```
+
+- xpu
+
+```bash
+python projects/Aquila/pipeline.py --model_path=/root/models/Aquila-7B --mode=huggingface --device=xpu
+```
+
+### 训练 
+- data preparation
 ```bash
 python projects/Aquila/utils/data_prepare.py
 ```