Skip to content

Commit ca21996

Browse files
committed
Remove old seamless class
1 parent 62aa064 commit ca21996

File tree

4 files changed

+4
-57
lines changed

4 files changed

+4
-57
lines changed

invokeai/app/invocations/denoise_latents.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939
from invokeai.backend.lora import LoRAModelRaw
4040
from invokeai.backend.model_manager import BaseModelType
4141
from invokeai.backend.model_patcher import ModelPatcher
42-
from invokeai.backend.stable_diffusion import PipelineIntermediateState, set_seamless
42+
from invokeai.backend.stable_diffusion import PipelineIntermediateState
4343
from invokeai.backend.stable_diffusion.denoise_context import DenoiseContext, DenoiseInputs
4444
from invokeai.backend.stable_diffusion.diffusers_pipeline import (
4545
ControlNetData,
@@ -920,7 +920,7 @@ def _lora_loader() -> Iterator[Tuple[LoRAModelRaw, float]]:
920920
ExitStack() as exit_stack,
921921
unet_info.model_on_device() as (model_state_dict, unet),
922922
ModelPatcher.apply_freeu(unet, self.unet.freeu_config),
923-
set_seamless(unet, self.unet.seamless_axes), # FIXME
923+
SeamlessExt.static_patch_model(unet, self.unet.seamless_axes), # FIXME
924924
# Apply the LoRA after unet has been moved to its target device for faster patching.
925925
ModelPatcher.apply_lora_unet(
926926
unet,

invokeai/app/invocations/latents_to_image.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
from invokeai.app.invocations.model import VAEField
2525
from invokeai.app.invocations.primitives import ImageOutput
2626
from invokeai.app.services.shared.invocation_context import InvocationContext
27-
from invokeai.backend.stable_diffusion import set_seamless
27+
from invokeai.backend.stable_diffusion.extensions.seamless import SeamlessExt
2828
from invokeai.backend.stable_diffusion.vae_tiling import patch_vae_tiling_params
2929
from invokeai.backend.util.devices import TorchDevice
3030

@@ -59,7 +59,7 @@ def invoke(self, context: InvocationContext) -> ImageOutput:
5959

6060
vae_info = context.models.load(self.vae.vae)
6161
assert isinstance(vae_info.model, (AutoencoderKL, AutoencoderTiny))
62-
with set_seamless(vae_info.model, self.vae.seamless_axes), vae_info as vae:
62+
with SeamlessExt.static_patch_model(vae_info.model, self.vae.seamless_axes), vae_info as vae:
6363
assert isinstance(vae, (AutoencoderKL, AutoencoderTiny))
6464
latents = latents.to(vae.device)
6565
if self.fp32:

invokeai/backend/stable_diffusion/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,9 @@
77
StableDiffusionGeneratorPipeline,
88
)
99
from invokeai.backend.stable_diffusion.diffusion import InvokeAIDiffuserComponent # noqa: F401
10-
from invokeai.backend.stable_diffusion.seamless import set_seamless # noqa: F401
1110

1211
__all__ = [
1312
"PipelineIntermediateState",
1413
"StableDiffusionGeneratorPipeline",
1514
"InvokeAIDiffuserComponent",
16-
"set_seamless",
1715
]

invokeai/backend/stable_diffusion/seamless.py

Lines changed: 0 additions & 51 deletions
This file was deleted.

0 commit comments

Comments
 (0)