From 125d783076e5bd9785beb05367a2d2566843a271 Mon Sep 17 00:00:00 2001 From: Ayan Das Date: Sun, 29 Jan 2023 12:23:18 +0000 Subject: [PATCH] fix typo in EMAModel's load_state_dict() (#2151) Possible typo introduced in 7c82a16fc14840429566aec40eb9e65aa57005fd --- src/diffusers/training_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/training_utils.py b/src/diffusers/training_utils.py index 7f43f553f6..c5449556a1 100644 --- a/src/diffusers/training_utils.py +++ b/src/diffusers/training_utils.py @@ -253,7 +253,7 @@ class EMAModel: if not isinstance(self.inv_gamma, (float, int)): raise ValueError("Invalid inv_gamma") - self.power = state_dict["power"].get("power", self.power) + self.power = state_dict.get("power", self.power) if not isinstance(self.power, (float, int)): raise ValueError("Invalid power")