Fix requirements and update README.“

3a1e6ddb · chenych · 66792d8d · 3a1e6ddb · 3a1e6ddb
Commit 3a1e6ddb authored Oct 22, 2025 by chenych
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 2 deletions

README.md README.md +4 -1

requirements.txt requirements.txt +3 -1

No files found.
--- a/README.md
+++ b/README.md
@@ -47,8 +47,11 @@ cd /your_code_path/deepseek-ocr_pytorch
 DTK: 25.04.1
 python: 3.10.12
 torch: 2.5.1+das.opt1.dtk25041
+torchvision: 0.19.1+das.opt1.dtk25041
+flash_attn: 2.6.1+das.opt1.dtk25041
+triton: 3.0.0+das.opt1.dtk25041
 transformers: 4.46.3
-vllm: 0.8.5
+vllm: 0.8.5.post1+das.opt4.dtk25041
 ```
 `Tips：以上dtk驱动、pytorch等DCU相关工具版本需要严格一一对应`, 其它非深度学习库参照requirements.txt安装：
 ```bash

--- a/requirements.txt
+++ b/requirements.txt
@@ -6,4 +6,6 @@ einops
 easydict
 addict
 Pillow
-numpy
+numpy==1.25.0
+pytest==8.4.1
+pytest-asyncio
\ No newline at end of file