jaidaken/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

New Year ruff cleanup. (#11595)

Browse Source
This commit is contained in:
comfyanonymous
2026-01-01 19:06:14 -08:00
committed by GitHub
parent 1bdc9a947f
commit 65cfcf5b1b
14 changed files with 35 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/modules/ema.py
View File

@@ -45,7 +45,7 @@ class LitEma(nn.Module):
shadow_params[sname] = shadow_params[sname].type_as(m_param[key])
shadow_params[sname].sub_(one_minus_decay * (shadow_params[sname] - m_param[key]))
else:
assert not key in self.m_name2s_name
assert key not in self.m_name2s_name
def copy_to(self, model):
m_param = dict(model.named_parameters())
@@ -54,7 +54,7 @@ class LitEma(nn.Module):
if m_param[key].requires_grad:
m_param[key].data.copy_(shadow_params[self.m_name2s_name[key]].data)
else:
assert not key in self.m_name2s_name
assert key not in self.m_name2s_name
def store(self, parameters):
"""