|
|
|
@ -538,6 +538,8 @@ class BasicTransformerBlock(nn.Module):
|
|
|
|
|
if "block" in transformer_options: |
|
|
|
|
block = transformer_options["block"] |
|
|
|
|
extra_options["block"] = block |
|
|
|
|
if "cond_or_uncond" in transformer_options: |
|
|
|
|
extra_options["cond_or_uncond"] = transformer_options["cond_or_uncond"] |
|
|
|
|
if "patches" in transformer_options: |
|
|
|
|
transformer_patches = transformer_options["patches"] |
|
|
|
|
else: |
|
|
|
|