chore: ruff

This commit is contained in:
psychedelicious 2024-03-12 01:54:51 +11:00
parent f69938c6a8
commit 60492500db
4 changed files with 3 additions and 10 deletions

View File

@ -83,9 +83,7 @@ class ModelManagerService(ModelManagerServiceBase):
logger=logger,
execution_device=execution_device,
)
convert_cache = ModelConvertCache(
cache_path=app_config.convert_cache_path, max_size=app_config.convert_cache
)
convert_cache = ModelConvertCache(cache_path=app_config.convert_cache_path, max_size=app_config.convert_cache)
loader = ModelLoadService(
app_config=app_config,
ram_cache=ram_cache,

View File

@ -4,7 +4,6 @@ wraps the safety_checker model. It respects the global "nsfw_checker"
configuration variable, that allows the checker to be supressed.
"""
import numpy as np
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
from PIL import Image

View File

@ -339,9 +339,7 @@ class InvokeAILogger(object): # noqa D102
loggers: Dict[str, logging.Logger] = {}
@classmethod
def get_logger(
cls, name: str = "InvokeAI", config: InvokeAIAppConfig = get_config()
) -> logging.Logger: # noqa D102
def get_logger(cls, name: str = "InvokeAI", config: InvokeAIAppConfig = get_config()) -> logging.Logger: # noqa D102
if name in cls.loggers:
return cls.loggers[name]

View File

@ -9,6 +9,4 @@ root_help = r"""Sets a root directory for the app. If omitted, the app will sear
app_arg_parser = ArgumentParser(description="Invoke Studio", formatter_class=RawTextHelpFormatter)
app_arg_parser.add_argument("--root", type=str, help=root_help)
app_arg_parser.add_argument(
"--version", action="version", version=__version__, help="Displays the version and exits."
)
app_arg_parser.add_argument("--version", action="version", version=__version__, help="Displays the version and exits.")