Controlnet/t2iadapter cleanup.

2025-09-14 13:35:05 +00:00 · 2023-08-21 23:20:49 -04:00
parent 763b0cf024
commit cf5ae46928
3 changed files with 58 additions and 55 deletions
--- a/comfy/t2i_adapter/adapter.py
+++ b/comfy/t2i_adapter/adapter.py
@@ -128,6 +128,8 @@ class Adapter(nn.Module):
            for j in range(self.nums_rb):
                idx = i * self.nums_rb + j
                x = self.body[idx](x)
+            features.append(None)
+            features.append(None)
            features.append(x)

        return features
@@ -259,6 +261,8 @@ class Adapter_light(nn.Module):
        features = []
        for i in range(len(self.channels)):
            x = self.body[i](x)
+            features.append(None)
+            features.append(None)
            features.append(x)

        return features