mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-21 13:48:46 +00:00
t2i_adapter: minor type annotation fix
This commit is contained in:
parent
9fbe86fbc9
commit
9b1e25e682
|
@ -25,7 +25,7 @@ class SDXLT2IAdapter(T2IAdapter[SDXLUNet]):
|
|||
)
|
||||
|
||||
def inject(self: "SDXLT2IAdapter", parent: fl.Chain | None = None) -> "SDXLT2IAdapter":
|
||||
def sanity_check_t2i(block: fl.Module) -> None:
|
||||
def sanity_check_t2i(block: fl.Chain) -> None:
|
||||
for t2i_layer in block.layers(layer_type=T2IFeatures):
|
||||
assert t2i_layer.name != self.name, f"T2I-Adapter named {self.name} is already injected"
|
||||
|
||||
|
|
Loading…
Reference in a new issue