mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
chore: ruff
This commit is contained in:
parent
f69938c6a8
commit
60492500db
@ -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,
|
||||
|
@ -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
|
||||
|
@ -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]
|
||||
|
||||
|
@ -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.")
|
||||
|
Loading…
Reference in New Issue
Block a user