Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: ruff

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/services/model_manager/model_manager_default.py
View File

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

1
invokeai/backend/image_util/safety_checker.py
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. configuration variable, that allows the checker to be supressed.
""" """
import numpy as np import numpy as np
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
from PIL import Image from PIL import Image

4
invokeai/backend/util/logging.py
View File

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

4
invokeai/frontend/cli/app_arg_parser.py
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 = ArgumentParser(description="Invoke Studio", formatter_class=RawTextHelpFormatter)
app_arg_parser.add_argument("--root", type=str, help=root_help) app_arg_parser.add_argument("--root", type=str, help=root_help)
app_arg_parser.add_argument( app_arg_parser.add_argument("--version", action="version", version=__version__, help="Displays the version and exits.")
"--version", action="version", version=__version__, help="Displays the version and exits."
)