Skip to content

Add AscendNPU support for Animatediff#335

Open
HelloWorldBeginner wants to merge 1 commit intoguoyww:mainfrom
HelloWorldBeginner:main
Open

Add AscendNPU support for Animatediff#335
HelloWorldBeginner wants to merge 1 commit intoguoyww:mainfrom
HelloWorldBeginner:main

Conversation

@HelloWorldBeginner
Copy link

I've modified some codes to add support for Ascend NPU, allowing Animatediff to train and inference on NPU.
It works fine on NPU.
NPU training
image

NPU Inference
image

@HelloWorldBeginner
Copy link
Author

Hi @guoyww
Can you review my codes?

@sguo112
Copy link

sguo112 commented Dec 16, 2024

thanks,let me try

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants