mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
7ea995149e
- Make environment variable settings case InSenSiTive: INVOKEAI_MAX_LOADED_MODELS and InvokeAI_Max_Loaded_Models environment variables will both set `max_loaded_models` - Updated realesrgan to use new config system. - Updated textual_inversion_training to use new config system. - Discovered a race condition when InvokeAIAppConfig is created at module load time, which makes it impossible to customize or replace the help message produced with --help on the command line. To fix this, moved all instances of get_invokeai_config() from module load time to object initialization time. Makes code cleaner, too. - Added `--from_file` argument to `invokeai-node-cli` and changed github action to match. CI tests will hopefully work now.
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
"""
|
|
This module defines a singleton object, "patchmatch" that
|
|
wraps the actual patchmatch object. It respects the global
|
|
"try_patchmatch" attribute, so that patchmatch loading can
|
|
be suppressed or deferred
|
|
"""
|
|
import numpy as np
|
|
import invokeai.backend.util.logging as logger
|
|
from invokeai.app.services.config import get_invokeai_config
|
|
|
|
class PatchMatch:
|
|
"""
|
|
Thin class wrapper around the patchmatch function.
|
|
"""
|
|
|
|
patch_match = None
|
|
tried_load: bool = False
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
@classmethod
|
|
def _load_patch_match(self):
|
|
config = get_invokeai_config()
|
|
if self.tried_load:
|
|
return
|
|
if config.try_patchmatch:
|
|
from patchmatch import patch_match as pm
|
|
|
|
if pm.patchmatch_available:
|
|
logger.info("Patchmatch initialized")
|
|
else:
|
|
logger.info("Patchmatch not loaded (nonfatal)")
|
|
self.patch_match = pm
|
|
else:
|
|
logger.info("Patchmatch loading disabled")
|
|
self.tried_load = True
|
|
|
|
@classmethod
|
|
def patchmatch_available(self) -> bool:
|
|
self._load_patch_match()
|
|
return self.patch_match and self.patch_match.patchmatch_available
|
|
|
|
@classmethod
|
|
def inpaint(self, *args, **kwargs) -> np.ndarray:
|
|
if self.patchmatch_available():
|
|
return self.patch_match.inpaint(*args, **kwargs)
|