chore: black fix

This commit is contained in:
blessedcoolant 2023-08-29 07:50:17 +12:00
parent 2a1d7342a7
commit 605e13eac0

View File

@ -29,10 +29,7 @@ def set_seamless(model: Union[UNet2DConditionModel, AutoencoderKL], seamless_axe
try: try:
to_restore = [] to_restore = []
#print("try seamless")
for m_name, m in model.named_modules(): for m_name, m in model.named_modules():
if isinstance(model, UNet2DConditionModel): if isinstance(model, UNet2DConditionModel):
if ".attentions." in m_name: if ".attentions." in m_name:
continue continue