style: rename
This commit is contained in:
@@ -2,15 +2,15 @@ import os
|
||||
|
||||
import folder_paths # pyright: ignore[reportMissingImports]
|
||||
|
||||
from ..domain.model import AutoencoderModel
|
||||
from ..service import CreateAutoencoderService
|
||||
from ..domain.model import Autoencoder
|
||||
from ..service import AutoencoderService
|
||||
|
||||
|
||||
class AutoencoderHandler:
|
||||
def __init__(self, create_autoencoder_service: CreateAutoencoderService) -> None:
|
||||
self.create_autoencoder_service = create_autoencoder_service
|
||||
def __init__(self, autoencoder_service: AutoencoderService) -> None:
|
||||
self.create_autoencoder_service = autoencoder_service
|
||||
|
||||
def create(self, checkpoint_name: str) -> AutoencoderModel:
|
||||
def create(self, checkpoint_name: str) -> Autoencoder:
|
||||
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}")
|
||||
|
||||
Reference in New Issue
Block a user