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

f7278638 · Will Berman · GitHub · b36cbd4f · f7278638
Unverified Commit f7278638 authored Mar 06, 2023 by Will Berman Committed by GitHub Mar 06, 2023
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 2 deletions

src/diffusers/training_utils.py src/diffusers/training_utils.py +0 -2

No files found.
--- a/src/diffusers/training_utils.py
+++ b/src/diffusers/training_utils.py
@@ -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.