fix typo in EMAModel's load_state_dict() (huggingface#2151)

Possible typo introduced in 7c82a16
nickkolok · Jan 29, 2023 · 125d783 · 125d783
1 parent fdf70cb
commit 125d783
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/training_utils.py b/src/diffusers/training_utils.py
@@ -253,7 +253,7 @@ def load_state_dict(self, state_dict: dict) -> None:
         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")