Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Suggested changes

Browse Source 
Co-Authored-By: Ryan Dick <14897797+RyanJDick@users.noreply.github.com>
This commit is contained in:
Sergey Borisov 2024-07-26 19:22:26 +03:00
parent ca21996a97
commit adf1a977ea
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
invokeai/backend/stable_diffusion/extensions/seamless.py
View File

@ -37,6 +37,9 @@ class SeamlessExt(ExtensionBase):
yield yield
return return
x_mode = "circular" if "x" in seamless_axes else "constant"
y_mode = "circular" if "y" in seamless_axes else "constant"
# override conv_forward # override conv_forward
# https://github.com/huggingface/diffusers/issues/556#issuecomment-1993287019 # https://github.com/huggingface/diffusers/issues/556#issuecomment-1993287019
def _conv_forward_asymmetric( def _conv_forward_asymmetric(
@ -51,18 +54,11 @@ class SeamlessExt(ExtensionBase):
) )
original_layers: List[Tuple[nn.Conv2d, Callable]] = [] original_layers: List[Tuple[nn.Conv2d, Callable]] = []
try: try:
x_mode = "circular" if "x" in seamless_axes else "constant" for layer in model.modules():
y_mode = "circular" if "y" in seamless_axes else "constant" if not isinstance(layer, torch.nn.Conv2d):
continue
conv_layers: List[torch.nn.Conv2d] = []
for module in model.modules():
if isinstance(module, torch.nn.Conv2d):
conv_layers.append(module)
for layer in conv_layers:
if isinstance(layer, LoRACompatibleConv) and layer.lora_layer is None: if isinstance(layer, LoRACompatibleConv) and layer.lora_layer is None:
layer.lora_layer = lambda *x: 0 layer.lora_layer = lambda *x: 0
original_layers.append((layer, layer._conv_forward)) original_layers.append((layer, layer._conv_forward))