diff --git a/README.md b/README.md new file mode 100644 index 0000000..fd25852 --- /dev/null +++ b/README.md @@ -0,0 +1,13 @@ +# enginex-hygon-vllm + +运行于【海光 DCU】系列算力卡的【文生视频】引擎,基于 diffusion 架构,支持 text-to-video-synthesis 等最新流行模型 + +因具体模型之间的启动方式和具体镜像会有略微差别,请详细查看 `/enginex` 目录下各个支持模型的启动测试方式。 + +## 可支持模型列表 +可在项目文件夹 `/enginex` 下查看具体可支持模型文件的运行方式。 + +支持模型列表: +- Shanghai_AI_Laboratory/animatediff +- AI-ModelScope/Open-Sora +- AI-ModelScope/Latte \ No newline at end of file diff --git a/enginex/Latte.md b/enginex/Latte.md new file mode 100644 index 0000000..7d3801d --- /dev/null +++ b/enginex/Latte.md @@ -0,0 +1,12 @@ +# 运行方式 + +```python +docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 + +docker run --shm-size 10g --network=host --name=latte --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal -v 项目地址(绝对路径):/home/ -it bash + +pip install -r requirements.txt +pip install timm --no-deps + +pip uninstall apex +``` diff --git a/enginex/Open-Sora.md b/enginex/Open-Sora.md new file mode 100644 index 0000000..0f57c57 --- /dev/null +++ b/enginex/Open-Sora.md @@ -0,0 +1,14 @@ +# 运行方式 + +```python +docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.3.0-ubuntu20.04-dtk24.04.2-py3.10 + +docker run --shm-size 10g --network=host --name=opensora --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v 项目地址(绝对路径):/home/ -v /opt/hyhal:/opt/hyhal:ro -it bash + +pip install -r requirements.txt + +pip install . + +pip install bitsandbytes-0.42.0+das.opt1.dtk24042-py3-none-any.whl [开发者社区](https://download.sourcefind.cn:65024/directlink/4/bitsandbytes/DAS1.2/bitsandbytes-0.42.0+das.opt1.dtk24042-py3-none-any.whl) +pip install diffusers-0.29.0+das.opt1.dtk24042-py3-none-any.whl [开发者社区](https://download.sourcefind.cn:65024/directlink/4/diffusers/DAS1.2/diffusers-0.29.0+das.opt1.dtk24042-py3-none-any.whl) +```