Commit 3a1e6ddb authored by chenych's avatar chenych
Browse files

Fix requirements and update README.“

parent 66792d8d
......@@ -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
......
......@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment