Controlnet refactor.

This commit is contained in:
comfyanonymous
2024-06-25 17:02:05 -04:00
parent 97b409cd48
commit 66aaa14001
4 changed files with 24 additions and 32 deletions

View File

@@ -153,7 +153,13 @@ class Adapter(nn.Module):
features.append(None)
features.append(x)
return features
features = features[::-1]
if self.xl:
return {"input": features[1:], "middle": features[:1]}
else:
return {"input": features}
class LayerNorm(nn.LayerNorm):
@@ -290,4 +296,4 @@ class Adapter_light(nn.Module):
features.append(None)
features.append(x)
return features
return {"input": features[::-1]}