Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 23:12:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

t2i_adapter: minor type annotation fix

Browse source
This commit is contained in:
Cédric Deltheil 2023-10-04 15:36:58 +02:00 committed by Cédric Deltheil
parent 9fbe86fbc9
commit 9b1e25e682
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/refiners/foundationals/latent_diffusion/stable_diffusion_xl/t2i_adapter.py
View file

@ -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"