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

fix(config): fix invisible_watermark handling

Browse Source 
This setting was hardcoded to True. Simplified logic around it to not have a conditional that does nothing.
This commit is contained in:
psychedelicious
2024-03-11 22:55:02 +11:00
parent 7ca447ded1
commit 22ac204678
1 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
invokeai/backend/image_util/invisible_watermark.py
View File

@ -10,9 +10,9 @@ from imwatermark import WatermarkEncoder
from PIL import Image from PIL import Image
import invokeai.backend.util.logging as logger import invokeai.backend.util.logging as logger
from invokeai.app.services.config import InvokeAIAppConfig from invokeai.app.services.config.config_default import get_config
config = InvokeAIAppConfig.get_config() config = get_config()
class InvisibleWatermark: class InvisibleWatermark:
@ -20,14 +20,8 @@ class InvisibleWatermark:
Wrapper around InvisibleWatermark module. Wrapper around InvisibleWatermark module.
""" """
@classmethod
def invisible_watermark_available(cls) -> bool:
return config.invisible_watermark
@classmethod @classmethod
def add_watermark(cls, image: Image.Image, watermark_text: str) -> Image.Image: def add_watermark(cls, image: Image.Image, watermark_text: str) -> Image.Image:
if not cls.invisible_watermark_available():
return image
logger.debug(f'Applying invisible watermark "{watermark_text}"') logger.debug(f'Applying invisible watermark "{watermark_text}"')
bgr = cv2.cvtColor(np.array(image.convert("RGB")), cv2.COLOR_RGB2BGR) bgr = cv2.cvtColor(np.array(image.convert("RGB")), cv2.COLOR_RGB2BGR)
encoder = WatermarkEncoder() encoder = WatermarkEncoder()