Skip to content
This repository was archived by the owner on Aug 6, 2025. It is now read-only.

Adapt to Ascend NPU#81

Open
qyliuAI wants to merge 1 commit intofacebookresearch:mainfrom
qyliuAI:main
Open

Adapt to Ascend NPU#81
qyliuAI wants to merge 1 commit intofacebookresearch:mainfrom
qyliuAI:main

Conversation

@qyliuAI
Copy link
Copy Markdown

@qyliuAI qyliuAI commented Apr 11, 2024

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

What I do in this commit

add NPU logic into:
1. train.py
2. sample.py
3. environment_npu.yml
4. utils/device_utils.py

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Apr 11, 2024
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants