New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[NPU] Contribute Ascend NPU Compatibility #348

Open

Emily0219 wants to merge 1 commit into hpcaitech:main from Emily0219:main

Emily0219 commented Apr 29, 2024

[NPU] Support STDiT2-XL/2 training and inference for Ascend NPU

What I do in this commit

opensora/models/layers/blocks.py
opensora/acceleration/shardformer/modeling/t5.py
opensora/utils/device_utils.py
requirements_npu.txt
scripts/inference-long.py
scripts/inference.py
scripts/train.py
setup.py
README.md
docs/zh_CN/README.md

Emily0219 force-pushed the main branch from 339f9e8 to 55e6d32 Compare

May 9, 2024 13:58


          Contribute Ascend NPU Compatibility

69121cf

Emily0219 force-pushed the main branch from 55e6d32 to 69121cf Compare

May 9, 2024 14:00

wuxibin89 commented Jul 15, 2024

Great! Is this PR valid for the latest v1.2?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet