Skip to content

GitLab

Unverified Commit 125d7830 authored by Ayan Das's avatar Ayan Das Committed by GitHub
Browse files

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

Possible typo introduced in 7c82a16f
parent fdf70cb5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/training_utils.py
View file @ 125d7830
...@@ -253,7 +253,7 @@ class EMAModel: ...@@ -253,7 +253,7 @@ class EMAModel:
if not isinstance(self.inv_gamma, (float, int)): if not isinstance(self.inv_gamma, (float, int)):
raise ValueError("Invalid inv_gamma") 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)): if not isinstance(self.power, (float, int)):
raise ValueError("Invalid power") raise ValueError("Invalid power")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment