Update README.md

README.md

@@ -36,6 +36,8 @@
 
     pip install -r requirements.txt
 
+    pip install .
+
 
 ### Docker（方法二）
 
@@ -52,6 +54,8 @@
 
     pip install -r requirements.txt
 
+    pip install .
+
 ### Anaconda (方法三)
 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装：
 https://developer.hpccube.com/tool/
@@ -61,8 +65,7 @@ https://developer.hpccube.com/tool/
     torch:2.1.0
     torchvision:0.16.0
     triton:2.1.0
-    apex:
-
+    apex:0.1
 
 Tips：以上dtk驱动、python、torch等DCU相关工具版本需要严格一一对应
 
@@ -74,6 +77,8 @@ Tips：以上dtk驱动、python、torch等DCU相关工具版本需要严格一
 
     pip install -r requirements.txt
 
+    pip install .
+
 ## 数据集
 
 完整数据集下载：https://drive.google.com/drive/folders/154S6raNg9NpDGQRlRhhAaYcAx5xq1Ok8
@@ -160,18 +165,12 @@ https://huggingface.co/DeepFloyd/t5-v1_1-xxl/tree/main  (T5)
 
 ### 命令行
 
-    # Sample 16x256x256 (5s/sample)
+    # Sample 16x256x256 (5s/sample) 显存 ~32G
     torchrun --standalone --nproc_per_node 1 scripts/inference.py configs/opensora/inference/16x256x256.py --ckpt-path ./path/to/your/ckpt.pth
 
-    # Sample 16x512x512 (20s/sample, 100 time steps)
+    # Sample 16x512x512 (20s/sample, 100 time steps) 显存 > 32G
     torchrun --standalone --nproc_per_node 1 scripts/inference.py configs/opensora/inference/16x512x512.py --ckpt-path ./path/to/your/ckpt.pth
 
-    # Sample 64x512x512 (40s/sample, 100 time steps)
-    torchrun --standalone --nproc_per_node 1 scripts/inference.py configs/opensora/inference/64x512x512.py --ckpt-path ./path/to/your/ckpt.pth
-
-    # Sample 64x512x512 with sequence parallelism (30s/sample, 100 time steps)
-    # sequence parallelism is enabled automatically when nproc_per_node is larger than 1
-    torchrun --standalone --nproc_per_node 2 scripts/inference.py configs/opensora/inference/64x512x512.py --ckpt-path ./path/to/your/ckpt.pth
 
 ## result