Skip to content

[bugfix] bugfix for npu free memory #9640

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

Merged
merged 11 commits into from
Oct 25, 2024
Merged

[bugfix] bugfix for npu free memory #9640

Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
e49d112
Improve NPU performance
Oct 11, 2024
d25d229
Improve NPU performance
Oct 12, 2024
9555cf3
Improve NPU performance
Oct 12, 2024
2b0febe
Improve NPU performance
Oct 14, 2024
0365ae4
Merge branch 'main' into main_npu
leisuzz Oct 14, 2024
458f68f
Merge branch 'huggingface:main' into main_npu
leisuzz Oct 24, 2024
5a742e6
[bugfix] bugfix for npu free memory
Oct 24, 2024
42271f1
[bugfix] bugfix for npu free memory
Oct 24, 2024
62a718c
[bugfix] bugfix for npu free memory
Oct 24, 2024
4ef65c9
Merge branch 'main' into main_npu
sayakpaul Oct 24, 2024
c4e509e
Merge branch 'main' into main_npu
leisuzz Oct 25, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/diffusers/training_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -284,7 +284,7 @@ def free_memory():
elif torch.backends.mps.is_available():
torch.mps.empty_cache()
elif is_torch_npu_available():
torch_npu.empty_cache()
torch_npu.npu.empty_cache()


# Adapted from torch-ema https://github.com/fadel/pytorch_ema/blob/master/torch_ema/ema.py#L14
Expand Down
Loading