feat: create vae loader

This commit is contained in:
Limitex
2025-10-31 20:53:51 +09:00
parent bcee329ca8
commit 1c96f63a4b
16 changed files with 422 additions and 9 deletions
+2 -1
View File
@@ -1,3 +1,4 @@
from .autoencoder_handler import AutoencoderHandler
from .pipeline_handler import PipelineHandler
__all__ = ["PipelineHandler"]
__all__ = ["PipelineHandler", "AutoencoderHandler"]
+17
View File
@@ -0,0 +1,17 @@
import os
import folder_paths # pyright: ignore[reportMissingImports]
from ..domain.model import AutoencoderModel
from ..service.create_autoencoder_service import CreateAutoencoderService
class AutoencoderHandler:
def __init__(self, create_autoencoder_service: CreateAutoencoderService) -> None:
self.create_autoencoder_service = create_autoencoder_service
def create(self, checkpoint_name: str) -> AutoencoderModel:
checkpoint_path = folder_paths.get_full_path("vae", checkpoint_name)
if not os.path.exists(checkpoint_path):
raise FileNotFoundError(f"VAE file not found: {checkpoint_name}")
return self.create_autoencoder_service.create_from_checkpoint(checkpoint_path)