mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-09-14 13:35:05 +00:00
Controlnet refactor.
This commit is contained in:
@@ -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]}
|
||||
|
Reference in New Issue
Block a user