AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

ema step, don't empty cuda cache (#2563)

Browse Source
This commit is contained in:
Will Berman
2023-03-06 01:54:56 -08:00
committed by GitHub
parent b36cbd4fba
commit f7278638e4
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/training_utils.py
View File

@@ -203,8 +203,6 @@ class EMAModel:
else:
s_param.copy_(param)
torch.cuda.empty_cache()
def copy_to(self, parameters: Iterable[torch.nn.Parameter]) -> None:
"""
Copy current averaged parameters into given collection of parameters.